Spaces:

zhenwusw
/

AnimeGAN

Runtime error

zhenwusw commited on Jan 26, 2022

Commit

3ad2675

1 Parent(s): 4ccaf3d

Change device to cpu

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,7 @@ model2 = torch.hub.load(
 model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
-    size=512, device="cuda",side_by_side=False
 )
 def inference(img, ver):
     if ver == 'version 2 (🔺 robustness,🔻 stylization)':

 model1 = torch.hub.load("AK391/animegan2-pytorch:main", "generator", pretrained="face_paint_512_v1",  device="cpu")
 face2paint = torch.hub.load(
     'AK391/animegan2-pytorch:main', 'face2paint',
+    size=512, device="cpu",side_by_side=False
 )
 def inference(img, ver):
     if ver == 'version 2 (🔺 robustness,🔻 stylization)':