Spaces:

dkdaniz
/

katara

Paused

Daniel Marques commited on Oct 15, 2023

Commit

3861b3b

•

1 Parent(s): e72e226

fix: add callback

Files changed (1) hide show

load_models.py CHANGED Viewed

@@ -215,7 +215,7 @@ def load_model(device_type, model_id, model_basename=None, LOGGING=logging, stre
         top_k=40,
         repetition_penalty=1.0,
         generation_config=generation_config,
-        streamer=streamer
         callbacks=[StreamingStdOutCallbackHandler()]
     )

         top_k=40,
         repetition_penalty=1.0,
         generation_config=generation_config,
+        streamer=streamer,
         callbacks=[StreamingStdOutCallbackHandler()]
     )