Spaces:

trysem
/

AnimeGANv2

Runtime error

culture commited on Jul 6, 2022

Commit

6283fb6

1 Parent(s): 00c32ac

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ model2 = torch.hub.load(
     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
-    device="cuda",
     progress=False
 )

     "AK391/animegan2-pytorch:main",
     "generator",
     pretrained=True,
+    device="cpu",
     progress=False
 )