ilotoki_thu
/
alpaca_lora_4bit
2
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add v1 model as default in custom monkey patch

This commit is contained in:
John Smith 2023-04-10 09:33:41 +08:00
parent f185b90c3e
commit 5d3267d80d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
text-generation-webui/custom_monkey_patch.py
View File

@ -16,7 +16,7 @@ def load_model_llama(*args, **kwargs):
print("Loading {} ...".format(model_path))
t0 = time.time()
model, tokenizer = load_llama_model_4bit_low_ram(config_path, model_path, groupsize=-1)
model, tokenizer = load_llama_model_4bit_low_ram(config_path, model_path, groupsize=-1, is_v1_model=True)
model = PeftModel.from_pretrained(model, lora_path, device_map={'': 0}, torch_dtype=torch.float32)
print('{} Lora Applied.'.format(lora_path))