Update demos/musicgen_colab.py
Browse files- demos/musicgen_colab.py +0 -2
demos/musicgen_colab.py
CHANGED
@@ -1,4 +1,3 @@
|
|
1 |
-
import spaces # <--- IMPORTANT: Add this import
|
2 |
import argparse
|
3 |
import logging
|
4 |
import os
|
@@ -236,7 +235,6 @@ class Predictor:
|
|
236 |
|
237 |
_default_model_name = "facebook/musicgen-melody"
|
238 |
|
239 |
-
@spaces.GPU(duration=90) # Use the decorator for Spaces
|
240 |
def predict_full(model, model_path, depth, use_mbd, text, melody, duration, topk, topp, temperature, cfg_coef):
|
241 |
# Initialize Predictor *INSIDE* the function
|
242 |
predictor = Predictor(model, depth)
|
|
|
|
|
1 |
import argparse
|
2 |
import logging
|
3 |
import os
|
|
|
235 |
|
236 |
_default_model_name = "facebook/musicgen-melody"
|
237 |
|
|
|
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)
|