Add files

app.py

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