Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 16:17:57 +01:00
Code Issues Packages Projects Releases 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
Show Stats 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...")