Explorar o código

Better recognize the 4chan model (for #19)

oobabooga %!s(int64=3) %!d(string=hai) anos
pai
achega
1e5e56fa2e
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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: