futranbg commited on
Commit
be2693c
·
1 Parent(s): 396b8d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -23,9 +23,9 @@ End of the document<|end|>
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,7 +40,7 @@ bloom_model_kwargs={
40
  "temperature": 0.01,
41
  # "truncate": 1512,
42
  "seed" : 42,
43
- "stop" : ["</s>","]]]]","<|end|>","]]]"],
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)