Hugging Face's logo

Spaces:
markqiu
/
prinvest_mate
Sleeping

App Files Community
Tuchuanhuhuhu commited on
Commit
a2154af
•
1 Parent(s): f4d58e4

bugfix: may show error for LlaMA on first launch

Browse files
Files changed (1) hide show
  1. modules/models/models.py +1 -1
modules/models/models.py CHANGED
@@ -32,7 +32,7 @@ def get_model(
32
  if model_type != ModelType.OpenAI:
33
  config.local_embedding = True
34
  # del current_model.model
35
- model = None
36
  chatbot = gr.Chatbot.update(label=model_name)
37
  try:
38
  if model_type == ModelType.OpenAI:
 
32
  if model_type != ModelType.OpenAI:
33
  config.local_embedding = True
34
  # del current_model.model
35
+ model = original_model
36
  chatbot = gr.Chatbot.update(label=model_name)
37
  try:
38
  if model_type == ModelType.OpenAI: