Don't include PeftModel every time

This commit is contained in:
oobabooga 2023-03-18 10:55:24 -03:00
parent 86b99006d9
commit 7c945cfe8e

View File

@ -1,13 +1,13 @@
from pathlib import Path
from peft import PeftModel
import modules.shared as shared
from modules.models import load_model
def add_lora_to_model(lora_name):
from peft import PeftModel
# Is there a more efficient way of returning to the base model?
if lora_name == "None":
print("Reloading the model to remove the LoRA...")