Spaces:
Runtime error
Runtime error
Michael Natanael
commited on
Commit
·
3402997
1
Parent(s):
12a4d4b
change max_new_tokens whisper
Browse files
app.py
CHANGED
@@ -72,7 +72,7 @@ pipe = pipeline(
|
|
72 |
batch_size=256, # batch size for inference - set based on your device
|
73 |
torch_dtype=torch_dtype,
|
74 |
device=device,
|
75 |
-
max_new_tokens=
|
76 |
return_timestamps=False, # Save memory
|
77 |
)
|
78 |
|
|
|
72 |
batch_size=256, # batch size for inference - set based on your device
|
73 |
torch_dtype=torch_dtype,
|
74 |
device=device,
|
75 |
+
max_new_tokens=128, # Limit text generation
|
76 |
return_timestamps=False, # Save memory
|
77 |
)
|
78 |
|