Merge pull request #172 from GuyAglionby/guyaglionby-fix-hf-kwargs

Save kwargs for huggingface TGI model

Author: arnavsinghvi11

dsp/modules/hf_client.py

@@ -32,6 +32,9 @@ def __init__(self, model, port, url="http://future-hgx-1", http_request_kwargs=N
         self.headers = {"Content-Type": "application/json"}
 
         self.kwargs = {
+            "model": model,
+            "port": port,
+            "url": url,
             "temperature": 0.01,
             "max_tokens": 75,
             "top_p": 0.97,