Spaces:

ThankGod
/

anime-gan

Running

ThankGod commited on Sep 11, 2022

Commit

9f94fce

•

1 Parent(s): cbb8f1f

update device

Files changed (1) hide show

app.py CHANGED Viewed

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

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