updated

app.py

@@ -115,8 +115,7 @@ def load_original_model():
         model = HFMusicgenForConditionalGeneration.from_pretrained(
             "facebook/musicgen-large",
             torch_dtype=torch.float16,
-            device=device,
-        )
+        ).to(device)
 
         model.eval()