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

Sort some imports

Browse source
This commit is contained in:
oobabooga 2023-06-25 01:44:36 -03:00
parent 365b672531
commit f0fcd1f697
8 changed files with 60 additions and 33 deletions
Download patch file Download diff file Expand all files Collapse all files

10
modules/monkey_patch_gptq_lora.py
View file

@ -7,10 +7,14 @@ sys.path.insert(0, str(Path("repositories/alpaca_lora_4bit")))
import autograd_4bit
from amp_wrapper import AMPWrapper
from autograd_4bit import (Autograd4bitQuantLinear,
load_llama_model_4bit_low_ram)
from autograd_4bit import (
Autograd4bitQuantLinear,
load_llama_model_4bit_low_ram
)
from monkeypatch.peft_tuners_lora_monkey_patch import (
Linear4bitLt, replace_peft_model_with_gptq_lora_model)
Linear4bitLt,
replace_peft_model_with_gptq_lora_model
)
from modules import shared
from modules.GPTQ_loader import find_quantized_model_file