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

Handle unloading LoRA from dropdown menu icon

Browse source
This commit is contained in:
oobabooga 2023-03-27 00:04:43 -03:00
parent b6e38e8b97
commit 3dc61284d5
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
modules/LoRA.py
View file

@ -18,11 +18,11 @@ def add_lora_to_model(lora_name):
# If a LoRA had been previously loaded, or if we want
# to unload a LoRA, reload the model
if shared.lora_name != "None" or lora_name == "None":
if shared.lora_name not in ['None', ''] or lora_name in ['None', '']:
reload_model()
shared.lora_name = lora_name
if lora_name != "None":
if lora_name not in ['None', '']:
print(f"Adding the LoRA {lora_name} to the model...")
params = {}
if not shared.args.cpu: