Luigi commited on
Commit
16c9738
·
1 Parent(s): fba2074

decrease batch number to save mem.

Browse files
Files changed (1) hide show
  1. 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=2048,
23
- n_ubatch=1024,
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