Spaces:

TheStageAI
/

Elastic-musicgen-large

Running on L40S

quazim commited on Jun 27

Commit

5471a60

1 Parent(s): 8c46cbe

added elastic model

Files changed (1) hide show

app.py CHANGED Viewed

@@ -55,7 +55,7 @@ def load_model():
     model = MusicgenForConditionalGeneration.from_pretrained(
                 "facebook/musicgen-large",
                 torch_dtype=torch.float16,
-                device=self.device,
                 mode="S",
                 __paged=True,
             )

     model = MusicgenForConditionalGeneration.from_pretrained(
                 "facebook/musicgen-large",
                 torch_dtype=torch.float16,
+                device="cuda",
                 mode="S",
                 __paged=True,
             )