final changes

app.py

@@ -25,7 +25,7 @@ from audiocraft.models import MusicGen
 
 MODEL = None  # Last used model
 IS_BATCHED = "facebook/MusicGen" in os.environ.get('SPACE_ID', '')
-MAX_BATCH_SIZE = 8
+MAX_BATCH_SIZE = 12
 BATCHED_DURATION = 15
 INTERRUPTING = False
 # We have to wrap subprocess call to clean a bit the log when using gr.make_waveform

tests/models/test_musicgen.py

@@ -13,7 +13,7 @@ from audiocraft.models import MusicGen
 class TestSEANetModel:
     def get_musicgen(self):
         mg = MusicGen.get_pretrained(name='debug', device='cpu')
-        mg.set_generation_params(duration=2.0, stride_extend=2.)
+        mg.set_generation_params(duration=2.0, extend_stride=2.)
         return mg
 
     def test_base(self):
@@ -52,7 +52,7 @@ class TestSEANetModel:
     def test_generate_long(self):
         mg = self.get_musicgen()
         mg.max_duration = 3.
-        mg.set_generation_params(duration=4., stride_extend=2.)
+        mg.set_generation_params(duration=4., extend_stride=2.)
         wav = mg.generate(
             ['youpi', 'lapin dort'])
         assert list(wav.shape) == [2, 1, 32000 * 4]