diff --git a/modules/LoRA.py b/modules/LoRA.py index 1d36c333..8c30e609 100644 --- a/modules/LoRA.py +++ b/modules/LoRA.py @@ -1,6 +1,7 @@ from pathlib import Path import torch +from peft import PeftModel import modules.shared as shared from modules.models import load_model @@ -14,8 +15,6 @@ def reload_model(): def add_lora_to_model(lora_name): - from peft import PeftModel - # If a LoRA had been previously loaded, or if we want # to unload a LoRA, reload the model if shared.lora_name not in ['None', ''] or lora_name in ['None', '']: