Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-09-20 18:45:09 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

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
Show stats 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 from pathlib import Path
import torch import torch
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
@ -14,8 +15,6 @@ def reload_model():
def add_lora_to_model(lora_name): def add_lora_to_model(lora_name):
from peft import PeftModel
# If a LoRA had been previously loaded, or if we want # If a LoRA had been previously loaded, or if we want
# to unload a LoRA, reload the model # to unload a LoRA, reload the model
if shared.lora_name not in ['None', ''] or lora_name in ['None', '']: if shared.lora_name not in ['None', ''] or lora_name in ['None', '']: