Spaces:

markqiu
/

prinvest_mate

Sleeping

markqiu commited on Dec 9, 2023

Commit

636fc2d

1 Parent(s): 5cb3963

fix: local_embedding error

Files changed (1) hide show

modules/models/models.py CHANGED Viewed

@@ -30,7 +30,7 @@ def get_model(
     lora_choices = ["No LoRA"]
     dont_change_lora_selector = False
     if model_type != ModelType.OpenAI:
-        config.local_embedding = True
     # del current_model.model
     model = original_model
     chatbot = gr.Chatbot.update(label=model_name)

     lora_choices = ["No LoRA"]
     dont_change_lora_selector = False
     if model_type != ModelType.OpenAI:
+        config['local_embedding'] = True
     # del current_model.model
     model = original_model
     chatbot = gr.Chatbot.update(label=model_name)