Spaces:

ai-forever
/

GHOST-2.0

Running on Zero

ai-forever commited on Feb 21

Commit

4a4650f

1 Parent(s): f99a7ba

cpu inpainting

Files changed (1) hide show

app.py CHANGED Viewed

@@ -146,7 +146,7 @@ if __name__ == "__main__":
     blender.eval()
     blender.cuda()
-    inpainter = LamaInpainter()
     app = FaceAnalysis(providers=['CUDAExecutionProvider'], allowed_modules=['detection'])
     app.prepare(ctx_id=0, det_size=(640, 640))

     blender.eval()
     blender.cuda()
+    inpainter = LamaInpainter('cpu')
     app = FaceAnalysis(providers=['CUDAExecutionProvider'], allowed_modules=['detection'])
     app.prepare(ctx_id=0, det_size=(640, 640))