hf-llm-api-pt

Running

radinhas commited on Dec 25, 2023

Commit

1385800

•

1 Parent(s): 746ab1f

Update apis/chat_api.py

Files changed (1) hide show

apis/chat_api.py CHANGED Viewed

@@ -170,7 +170,7 @@ class ChatAPIApp:
         "default": "t5-base",
         }
         if item.model in MODEL_MAP.keys():
-            target_model = MODEL_MAP[item.model]
         else:
             target_model = "default"

         "default": "t5-base",
         }
         if item.model in MODEL_MAP.keys():
+            target_model = item.model
         else:
             target_model = "default"