Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
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 =
|
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(
|
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 |
|