Remove unused imports

2023-03-16 10:18:34 -03:00
parent a577fb1077
commit 1c378965e1
2 changed files with 3 additions and 4 deletions
--- a/modules/models.py
+++ b/modules/models.py
@@ -16,8 +16,7 @@ transformers.logging.set_verbosity_error()
 local_rank = None

 if shared.args.flexgen:
-    from flexgen.flex_opt import (CompressionConfig, ExecutionEnv, OptLM,
-                                  Policy, str2bool)
+    from flexgen.flex_opt import CompressionConfig, ExecutionEnv, OptLM, Policy

 if shared.args.deepspeed:
    import deepspeed