Merge remote-tracking branch 'hf2/main' into our_hf2
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 |
|