Spaces:

shi-labs
/

Smooth-Diffusion

Runtime error

JiayiGuo821 commited on Mar 22

Commit

302fb80

•

1 Parent(s): a534a2d

Add files

Files changed (1) hide show

app.py CHANGED Viewed

@@ -256,7 +256,8 @@ class wrapper(object):
                  tag_lora=None,
                  tag_scheduler=None,):
-        self.device = "cuda"
         if fp16:
             self.torch_dtype = torch.float16
         else:

                  tag_lora=None,
                  tag_scheduler=None,):
+        # self.device = "cuda"
+        self.device = "cpu"
         if fp16:
             self.torch_dtype = torch.float16
         else: