Spaces:
Running
on
Zero
Running
on
Zero
revise app.py with 512 new tokens
Browse files
app.py
CHANGED
@@ -102,7 +102,7 @@ Only return the transcription in Thai language."""
|
|
102 |
with torch.no_grad():
|
103 |
outputs = model.generate(
|
104 |
**inputs,
|
105 |
-
max_new_tokens=256
|
106 |
do_sample=False,
|
107 |
pad_token_id=processor.tokenizer.pad_token_id
|
108 |
)
|
|
|
102 |
with torch.no_grad():
|
103 |
outputs = model.generate(
|
104 |
**inputs,
|
105 |
+
max_new_tokens=512, ##256
|
106 |
do_sample=False,
|
107 |
pad_token_id=processor.tokenizer.pad_token_id
|
108 |
)
|