Update app.py

app.py

@@ -86,8 +86,8 @@ opt = parser.parse_args()
 
 ### Global variables and data structures
 
-device = f'cuda:{opt.device}' if opt.device >= 0 else 'cpu'
-
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+print(device)
 
 if opt.model is None:
     model_dict = {
@@ -98,9 +98,8 @@ else:
         opt.model = os.path.abspath(os.path.join('checkpoints', opt.model))
     model_dict = {os.path.splitext(os.path.basename(opt.model))[0]: opt.model}
 
-dtype = torch.float32 if device == 'cpu' else torch.float16
 models = {
-    k: StableMultiDiffusion3Pipeline(device, dtype=dtype, hf_key=v, has_i2t=False)
+    k: StableMultiDiffusion3Pipeline(device, hf_key=v, has_i2t=False).cuda()
     for k, v in model_dict.items()
 }