ddyuudd commited on
Commit
e6f6d64
·
verified ·
1 Parent(s): 59e7707

Fixed the test.py to be compatible with older transformers versions

Browse files
Files changed (1) hide show
  1. test.py +1 -1
test.py CHANGED
@@ -13,7 +13,7 @@ src_text = "🐈はとてもかわいいの。おててがまるくてふわふ
13
 
14
  user_input = [{"role": "user", "content": prompt.format(src_lang=src_lang, tgt_lang=tgt_lang, src_text=src_text)}]
15
 
16
- response = chat_pipeline(user_input)
17
 
18
  print("-" * 20)
19
  print("Source Text:")
 
13
 
14
  user_input = [{"role": "user", "content": prompt.format(src_lang=src_lang, tgt_lang=tgt_lang, src_text=src_text)}]
15
 
16
+ response = chat_pipeline(user_input, max_new_tokens=512)
17
 
18
  print("-" * 20)
19
  print("Source Text:")