Spaces:
Running
Running
decrease batch number to save mem.
Browse files- summarize_transcript.py +2 -3
summarize_transcript.py
CHANGED
|
@@ -19,11 +19,10 @@ def load_model(repo_id, filename):
|
|
| 19 |
seed=1337,
|
| 20 |
n_ctx=32768, # Context size
|
| 21 |
verbose=False, # Reduced verbosity for cleaner output
|
| 22 |
-
n_batch=
|
| 23 |
-
n_ubatch=
|
| 24 |
v_type=2,
|
| 25 |
k_type=2,
|
| 26 |
-
n_threads=6
|
| 27 |
)
|
| 28 |
|
| 29 |
return llm
|
|
|
|
| 19 |
seed=1337,
|
| 20 |
n_ctx=32768, # Context size
|
| 21 |
verbose=False, # Reduced verbosity for cleaner output
|
| 22 |
+
n_batch=512,
|
| 23 |
+
n_ubatch=256,
|
| 24 |
v_type=2,
|
| 25 |
k_type=2,
|
|
|
|
| 26 |
)
|
| 27 |
|
| 28 |
return llm
|