Fixed the test.py to be compatible with older transformers versions
Browse files
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:")
|