AI/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-10-01 01:26:03 -04:00
Code Issues Packages Projects Releases Wiki Activity

Remove unused import

Browse Source
This commit is contained in:
oobabooga 2023-03-17 11:43:11 -03:00
parent a717fd709d
commit 614dad0075
2 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
modules/LoRA.py
View File

@ -10,7 +10,7 @@ def add_lora_to_model(lora_name):
# Is there a more efficient way of returning to the base model? # Is there a more efficient way of returning to the base model?
if lora_name == "None": if lora_name == "None":
print(f"Reloading the model to remove the LoRA...") print("Reloading the model to remove the LoRA...")
shared.model, shared.tokenizer = load_model(shared.model_name) shared.model, shared.tokenizer = load_model(shared.model_name)
else: else:
print(f"Adding the LoRA {lora_name} to the model...") print(f"Adding the LoRA {lora_name} to the model...")

1
modules/models.py
View File

@ -8,7 +8,6 @@ import numpy as np
import torch import torch
import transformers import transformers
from accelerate import infer_auto_device_map, init_empty_weights from accelerate import infer_auto_device_map, init_empty_weights
from peft import PeftModel
from transformers import (AutoConfig, AutoModelForCausalLM, AutoTokenizer, from transformers import (AutoConfig, AutoModelForCausalLM, AutoTokenizer,
BitsAndBytesConfig) BitsAndBytesConfig)