Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases 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
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', '']: