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