Spaces:

alvanlii
/

RDM-Region-Aware-Diffusion-Model

Runtime error

alvanlii commited on Jan 31, 2023

Commit

c83cce1

1 Parent(s): f2fccf5

Update run_edit.py

Files changed (1) hide show

run_edit.py CHANGED Viewed

@@ -45,7 +45,7 @@ import cv2
 def str2bool(x):
     return x.lower() in ('true')
-USE_CPU = True
 device = torch.device('cuda:0' if (torch.cuda.is_available() and not USE_CPU) else 'cpu')

 def str2bool(x):
     return x.lower() in ('true')
+USE_CPU = False
 device = torch.device('cuda:0' if (torch.cuda.is_available() and not USE_CPU) else 'cpu')