Spaces:

sweetcocoa
/

pop2piano

Sleeping

sweetcocoa commited on Nov 5, 2022

Commit

1aa2e4c

•

1 Parent(s): 11b1c3b

only cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,8 @@ def model_load():
     wrapper = wrapper.load_from_checkpoint(
         "https://huggingface.co/sweetcocoa/pop2piano/resolve/main/model-1999-val_0.67311615.ckpt",
         config=config,
-    ).cuda()
     model_id = "dpipqxiy"
     wrapper.eval()
     return wrapper, model_id, config

     wrapper = wrapper.load_from_checkpoint(
         "https://huggingface.co/sweetcocoa/pop2piano/resolve/main/model-1999-val_0.67311615.ckpt",
         config=config,
+        map_location="cpu",
+    )
     model_id = "dpipqxiy"
     wrapper.eval()
     return wrapper, model_id, config