futranbg commited on
Commit
119eed3
·
1 Parent(s): 611499f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -30,10 +30,17 @@ End of the texts
30
  {target} translated texts:
31
 
32
  """
33
-
34
  model_kwargs={
35
  "max_new_tokens":2048,
36
  "temperature": 0.01,
 
 
 
 
 
 
 
 
37
  "truncate": 1000,
38
  "seed" : 256,
39
  "stop" : ["</s>","<|endoftext|>","<|end|>"],
@@ -41,7 +48,7 @@ model_kwargs={
41
 
42
  llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
43
  llm2 = HuggingFaceHub(repo_id=starchat_repo, task="text-generation", model_kwargs=model_kwargs)
44
- llm3 = HuggingFaceHub(repo_id=bloom_repo, task="text-generation", model_kwargs=model_kwargs)
45
 
46
  def translation(source, target, text):
47
  response = text
 
30
  {target} translated texts:
31
 
32
  """
 
33
  model_kwargs={
34
  "max_new_tokens":2048,
35
  "temperature": 0.01,
36
+ "truncate": 4096,
37
+ "seed" : 256,
38
+ "stop" : ["</s>","<|endoftext|>","<|end|>"],
39
+ }
40
+
41
+ bloom_model_kwargs={
42
+ "max_new_tokens":1000,
43
+ "temperature": 0.01,
44
  "truncate": 1000,
45
  "seed" : 256,
46
  "stop" : ["</s>","<|endoftext|>","<|end|>"],
 
48
 
49
  llm1 = HuggingFaceHub(repo_id=llama_repo, task="text-generation", model_kwargs=model_kwargs)
50
  llm2 = HuggingFaceHub(repo_id=starchat_repo, task="text-generation", model_kwargs=model_kwargs)
51
+ llm3 = HuggingFaceHub(repo_id=bloom_repo, task="text-generation", model_kwargs=bloom_model_kwargs)
52
 
53
  def translation(source, target, text):
54
  response = text