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

Move an import

Browse source
This commit is contained in:
oobabooga 2023-03-29 22:50:58 -03:00
parent 55755e27b9
commit a21e580782
1 changed files with 1 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
modules/LoRA.py
View file

@ -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', '']: