iShare commited on
Commit
49c7d9d
·
1 Parent(s): 0d6b068

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -51,9 +51,9 @@ api_url = f"https://api-inference.huggingface.co/pipeline/feature-extraction/{mo
51
  headers = {"Authorization": f"Bearer {hf_token}"}
52
 
53
  llm = HuggingFaceHub(repo_id=repo_id,
54
- model_kwargs={"min_length":100,
55
  "max_new_tokens":1024, "do_sample":True,
56
- "temperature":0.1,
57
  "top_k":50,
58
  "top_p":0.95, "eos_token_id":49155})
59
 
 
51
  headers = {"Authorization": f"Bearer {hf_token}"}
52
 
53
  llm = HuggingFaceHub(repo_id=repo_id,
54
+ model_kwargs={"min_length":512,
55
  "max_new_tokens":1024, "do_sample":True,
56
+ "temperature":0.01,
57
  "top_k":50,
58
  "top_p":0.95, "eos_token_id":49155})
59