Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -23,9 +23,9 @@ End of the document<|end|>
|
|
23 |
|
24 |
bloom_template = """Text translation.
|
25 |
{source} text:
|
26 |
-
|
27 |
{target} translated text:
|
28 |
-
|
29 |
|
30 |
model_kwargs={
|
31 |
"max_new_tokens":2048,
|
@@ -40,7 +40,7 @@ bloom_model_kwargs={
|
|
40 |
"temperature": 0.01,
|
41 |
# "truncate": 1512,
|
42 |
"seed" : 42,
|
43 |
-
"stop" : ["</s>","
|
44 |
}
|
45 |
|
46 |
llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
|
|
|
23 |
|
24 |
bloom_template = """Text translation.
|
25 |
{source} text:
|
26 |
+
<|beginoftext|>{query}<|endoftext|>
|
27 |
{target} translated text:
|
28 |
+
<|beginoftext|>"""
|
29 |
|
30 |
model_kwargs={
|
31 |
"max_new_tokens":2048,
|
|
|
40 |
"temperature": 0.01,
|
41 |
# "truncate": 1512,
|
42 |
"seed" : 42,
|
43 |
+
"stop" : ["</s>","<|endoftext|>","<|end|>"],
|
44 |
}
|
45 |
|
46 |
llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
|