Spaces:

shi-labs
/

Smooth-Diffusion

Sleeping

JiayiGuo821 commited on Mar 25

Commit

a698371

•

1 Parent(s): d55ad57

Add files

Files changed (1) hide show

app.py CHANGED Viewed

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

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