fixed missing moving model to device

PersianC4t · Feb 12, 2020 · b49a679 · b49a679
1 parent 9c3a561
commit b49a679
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/demucs/separate.py b/demucs/separate.py
@@ -134,7 +134,7 @@ def main():
         download_file(url, model_path)
     if sha256 is not None:
         verify_file(model_path, sha256)
-    model = load_model(model_path)
+    model = load_model(model_path).to(args.device)
     out = args.out / args.name
     out.mkdir(parents=True, exist_ok=True)
     source_names = ["drums", "bass", "other", "vocals"]