Spaces:

ai4bharat
/

IndicTrans3-beta

Running on Zero

AshwinSankar commited on Jun 18

Commit

bf67d50

verified ·

1 Parent(s): 24bcef0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -171,7 +171,6 @@ def translate_message(
             "num_beams": 1,
             "repetition_penalty": repetition_penalty,
             "use_cache": True,  # Enable KV cache
-            "pad_token_id": tokenizer.eos_token_id,
         }
         t = Thread(target=model.generate, kwargs=generate_kwargs)

             "num_beams": 1,
             "repetition_penalty": repetition_penalty,
             "use_cache": True,  # Enable KV cache
         }
         t = Thread(target=model.generate, kwargs=generate_kwargs)