continuation with absolute duration

sakemin · Oct 18, 2023 · b26dac8 · b26dac8
1 parent 67a5c92
commit b26dac8
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/predict.py b/predict.py
@@ -355,15 +355,7 @@ def predict(
                 audio_chords_duration = audio_chords_wavform.shape[-1] / sr
 
                 if continuation: 
-                    if (
-                        duration + audio_chords_duration
-                        > model.lm.cfg.dataset.segment_duration
-                    ):
-                        raise ValueError(
-                            "duration + continuation duration must be <= 30 seconds"
-                        )
-
-                    set_generation_params(duration + audio_chords_duration)
+                    set_generation_params(duration)
 
                     if text_chords is None or text_chords == '': # Case 6
                         wav, tokens  = model.generate_continuation(