adefossez commited on
Commit
7bcf4ee
2 Parent(s): 4388bc6 34441c8

Merge remote-tracking branch 'hf2/main' into our_hf2

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -25,7 +25,7 @@ from audiocraft.models import MusicGen
25
 
26
  MODEL = None # Last used model
27
  IS_BATCHED = "facebook/MusicGen" in os.environ.get('SPACE_ID', '')
28
- MAX_BATCH_SIZE = 12
29
  BATCHED_DURATION = 15
30
  INTERRUPTING = False
31
  # We have to wrap subprocess call to clean a bit the log when using gr.make_waveform
@@ -41,7 +41,7 @@ def _call_nostderr(*args, **kwargs):
41
 
42
  sp.call = _call_nostderr
43
  # Preallocating the pool of processes.
44
- pool = ProcessPoolExecutor(4)
45
  pool.__enter__()
46
 
47
 
 
25
 
26
  MODEL = None # Last used model
27
  IS_BATCHED = "facebook/MusicGen" in os.environ.get('SPACE_ID', '')
28
+ MAX_BATCH_SIZE = 6
29
  BATCHED_DURATION = 15
30
  INTERRUPTING = False
31
  # We have to wrap subprocess call to clean a bit the log when using gr.make_waveform
 
41
 
42
  sp.call = _call_nostderr
43
  # Preallocating the pool of processes.
44
+ pool = ProcessPoolExecutor(3)
45
  pool.__enter__()
46
 
47