Spaces:

dkdaniz
/

katara

Paused

Daniel Marques commited on Oct 15, 2023

Commit

6968cd1

•

1 Parent(s): c4cd469

fix: add types

Files changed (1) hide show

load_models.py CHANGED Viewed

@@ -225,7 +225,7 @@ def load_model(device_type, model_id, model_basename=None, LOGGING=logging, stre
         repetition_penalty=1.0,
         generation_config=generation_config,
         streamer=streamer,
-        callbacks=[MyCustomSyncHandler(), MyCustomAsyncHandler()]
     )
     local_llm = HuggingFacePipeline(pipeline=pipe)

         repetition_penalty=1.0,
         generation_config=generation_config,
         streamer=streamer,
+        # callbacks=callbacks
     )
     local_llm = HuggingFacePipeline(pipeline=pipe)