Spaces:
Running
Running
use default n_batch & n_ubatch
Browse files- summarize_transcript.py +0 -2
summarize_transcript.py
CHANGED
|
@@ -19,8 +19,6 @@ def load_model(repo_id, filename, cpu_only=False):
|
|
| 19 |
seed=1337,
|
| 20 |
n_ctx=32768, # Context size
|
| 21 |
verbose=True, # Reduced verbosity for cleaner output
|
| 22 |
-
n_batch=512,
|
| 23 |
-
n_ubatch=256,
|
| 24 |
v_type=2,
|
| 25 |
k_type=2,
|
| 26 |
)
|
|
|
|
| 19 |
seed=1337,
|
| 20 |
n_ctx=32768, # Context size
|
| 21 |
verbose=True, # Reduced verbosity for cleaner output
|
|
|
|
|
|
|
| 22 |
v_type=2,
|
| 23 |
k_type=2,
|
| 24 |
)
|