diff --git a/modules/LoRA.py b/modules/LoRA.py index b568e57b..f29523d2 100644 --- a/modules/LoRA.py +++ b/modules/LoRA.py @@ -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...")