Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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":
|
55 |
"max_new_tokens":1024, "do_sample":True,
|
56 |
-
"temperature":0.
|
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 |
|