zetavg commited on
Commit
701dba0
1 Parent(s): e615353
Files changed (1) hide show
  1. llama_lora/ui/inference_ui.py +2 -3
llama_lora/ui/inference_ui.py CHANGED
@@ -53,9 +53,8 @@ def do_inference(
53
  yield message, '[0]'
54
  return
55
 
56
- if lora_model_name == "None":
57
- model = get_base_model()
58
- else:
59
  model = get_model_with_lora(lora_model_name)
60
  tokenizer = get_tokenizer()
61
 
 
53
  yield message, '[0]'
54
  return
55
 
56
+ model = get_base_model()
57
+ if not lora_model_name == "None":
 
58
  model = get_model_with_lora(lora_model_name)
59
  tokenizer = get_tokenizer()
60