ford442 commited on
Commit
a644c95
·
verified ·
1 Parent(s): 8651fc9

Update demos/musicgen_colab.py

Browse files
Files changed (1) hide show
  1. 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)