Update demos/musicgen_colab.py
Browse files- demos/musicgen_colab.py +1 -1
demos/musicgen_colab.py
CHANGED
|
@@ -238,7 +238,7 @@ _default_model_name = "facebook/musicgen-melody"
|
|
| 238 |
def predict_full(model, model_path, depth, use_mbd, text, melody, duration, topk, topp, temperature, cfg_coef):
|
| 239 |
# Initialize Predictor *INSIDE* the function
|
| 240 |
predictor = Predictor(model, depth)
|
| 241 |
-
|
| 242 |
text=text,
|
| 243 |
melody=melody,
|
| 244 |
duration=duration,
|
|
|
|
| 238 |
def predict_full(model, model_path, depth, use_mbd, text, melody, duration, topk, topp, temperature, cfg_coef):
|
| 239 |
# Initialize Predictor *INSIDE* the function
|
| 240 |
predictor = Predictor(model, depth)
|
| 241 |
+
wav, diffusion_wav = predictor.predict( # Unpack directly!
|
| 242 |
text=text,
|
| 243 |
melody=melody,
|
| 244 |
duration=duration,
|