Procházet zdrojové kódy

Better recognize the 4chan model (for #19)

oobabooga před 3 roky
rodič
revize
1e5e56fa2e
1 změnil soubory, kde provedl 1 přidání a 1 odebrání
  1. 1 1
      server.py

+ 1 - 1
server.py

@@ -155,7 +155,7 @@ def formatted_outputs(reply, model_name):
         if model_name.lower().startswith('galactica'):
             reply = fix_galactica(reply)
             return reply, reply, generate_basic_html(reply)
-        elif model_name.lower().startswith('gpt4chan'):
+        elif model_name.lower().startswith(('gpt4chan', 'gpt-4chan', '4chan')):
             reply = fix_gpt4chan(reply)
             return reply, 'Only applicable for GALACTICA models.', generate_4chan_html(reply)
         else: