Auto-deploy from GitHub: e1156fe13977a8a88d0d70832354b216355882c4

src/musiclime/factorization.py

@@ -141,7 +141,9 @@ class OpenUnmixFactorization:
             device=device,
         )
 
-        components = [prediction[key][0].mean(dim=0).numpy() for key in prediction]
+        components = [
+            prediction[key][0].mean(dim=0).cpu().numpy() for key in prediction
+        ]
         names = list(prediction.keys())
 
         return components, names