Spaces:

oguzakif
/

video-object-remover

Runtime error

oguzakif commited on Apr 19, 2023

Commit

6c89e77

1 Parent(s): e7aba9f

torch device selector added

Files changed (1) hide show

FGT_codes/tool/video_inpainting.py CHANGED Viewed

@@ -415,7 +415,8 @@ def save_results(outdir, comp_frames):
 def video_inpainting(args, imgArr, imgMaskArr):
-    device = torch.device("cuda:{}".format(args.gpu))
     print(args)
     if args.opt is not None:
         with open(args.opt, "r") as f:

 def video_inpainting(args, imgArr, imgMaskArr):
+    #device = torch.device("cuda:{}".format(args.gpu))
+    device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
     print(args)
     if args.opt is not None:
         with open(args.opt, "r") as f: