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

Sort the imports

Browse source
This commit is contained in:
oobabooga 2023-03-17 11:42:25 -03:00
parent 7d97287e69
commit a717fd709d
4 changed files with 5 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/models.py
View file

@ -8,11 +8,10 @@ import numpy as np
import torch
import transformers
from accelerate import infer_auto_device_map, init_empty_weights
from peft import PeftModel
from transformers import (AutoConfig, AutoModelForCausalLM, AutoTokenizer,
BitsAndBytesConfig)
from peft import PeftModel
import modules.shared as shared
transformers.logging.set_verbosity_error()