Luigi commited on
Commit
6add551
·
1 Parent(s): 83a3f98

use default n_batch & n_ubatch

Browse files
Files changed (1) hide show
  1. 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
  )