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

Don't include PeftModel every time

Browse source
This commit is contained in:
oobabooga 2023-03-18 10:55:24 -03:00
parent 86b99006d9
commit 7c945cfe8e
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/LoRA.py
View file

@ -1,13 +1,13 @@
from pathlib import Path from pathlib import Path
from peft import PeftModel
import modules.shared as shared import modules.shared as shared
from modules.models import load_model from modules.models import load_model
def add_lora_to_model(lora_name): def add_lora_to_model(lora_name):
from peft import PeftModel
# 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("Reloading the model to remove the LoRA...") print("Reloading the model to remove the LoRA...")