Spaces:

dfuhge
/

Animate_SVG_v2

Running

Daniel Gil-U Fuhge commited on Mar 16

Commit

163b3a3

•

1 Parent(s): 347fa2e

change device to cpu

Files changed (1) hide show

animationPipeline.py CHANGED Viewed

@@ -27,7 +27,7 @@ def animateLogo(path : str):
         use_positional_encoder=True
     ).to(device)
-    model.load_state_dict(torch.load("models/animation_transformer.pth"))
     df = compute_embedding(path, "src/preprocessing/deepsvg/deepsvg_models/deepSVG_hierarchical_ordered.pth.tar")
     df = df.drop("animation_id", axis=1)

         use_positional_encoder=True
     ).to(device)
+    model.load_state_dict(torch.load("models/animation_transformer.pth", map_location=torch.device('cpu')))
     df = compute_embedding(path, "src/preprocessing/deepsvg/deepsvg_models/deepSVG_hierarchical_ordered.pth.tar")
     df = df.drop("animation_id", axis=1)