Преглед на файлове

Don't load the model twice when using --lora

oobabooga преди 2 години
родител
ревизия
d8e950d6bd
променени са 1 файла, в които са добавени 1 реда и са изтрити 3 реда
  1. 1 3
      server.py

+ 1 - 3
server.py

@@ -233,9 +233,7 @@ else:
     shared.model_name = available_models[i]
 shared.model, shared.tokenizer = load_model(shared.model_name)
 if shared.args.lora:
-    print(shared.args.lora)
-    shared.lora_name = shared.args.lora
-    add_lora_to_model(shared.lora_name)
+    add_lora_to_model(shared.args.lora)
 
 # Default UI settings
 default_preset = shared.settings['presets'][next((k for k in shared.settings['presets'] if re.match(k.lower(), shared.model_name.lower())), 'default')]