Update networks/huggingface_streamer.py

networks/huggingface_streamer.py

@@ -37,6 +37,7 @@ class HuggingfaceStreamer:
         max_new_tokens: int = None,
         api_key: str = None,
         use_cache: bool = False,
+        **kwargs,
     ):
         # https://huggingface.co/docs/api-inference/detailed_parameters?code=curl
         # curl --proxy http://<server>:<port> https://api-inference.huggingface.co/models/<org>/<model_name> -X POST -d '{"inputs":"who are you?","parameters":{"max_new_token":64}}' -H 'Content-Type: application/json' -H 'Authorization: Bearer <HF_TOKEN>'