futranbg commited on
Commit
9b71ff4
·
1 Parent(s): 69acd7a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -6
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)