mmarch/text-generation-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Remove unused imports

Browse source
This commit is contained in:
oobabooga 2023-03-16 10:18:34 -03:00
parent a577fb1077
commit 1c378965e1
2 changed files with 3 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/models.py
View file

@ -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