Spaces:

DroolingPanda
/

teachingAssistant

Running

Michael Hu commited on Jan 26

Commit

77b7581

1 Parent(s): 2f1e419

fix bug

Files changed (1) hide show

utils/translation.py CHANGED Viewed

@@ -48,7 +48,7 @@ def translate_text(text):
             # Generate translation with target language specification
             outputs = model.generate(
                 **inputs,
-                forced_bos_token_id=tokenizer.convert_tokens_to_ids["zho_Hans"],
                 max_new_tokens=1024
             )

             # Generate translation with target language specification
             outputs = model.generate(
                 **inputs,
+                forced_bos_token_id=tokenizer.convert_tokens_to_ids("zho_Hans"),
                 max_new_tokens=1024
             )