Procházet zdrojové kódy

Fix the API extension

oobabooga před 2 roky
rodič
revize
64f5c90ee7
1 změnil soubory, kde provedl 1 přidání a 0 odebrání
  1. 1 0
      extensions/api/script.py

+ 1 - 0
extensions/api/script.py

@@ -57,6 +57,7 @@ class Handler(BaseHTTPRequestHandler):
                 'length_penalty': float(body.get('length_penalty', 1)),
                 'early_stopping': bool(body.get('early_stopping', False)),
                 'seed': int(body.get('seed', -1)),
+                'add_bos_token': int(body.get('add_bos_token', True)),
             }
 
             generator = generate_reply(