Add device and dtype logging in app.py for better debugging

app.py

@@ -188,7 +188,7 @@ def load_infinity(
     model_path='', 
     scale_schedule=None, 
     vae=None, 
-    device='cuda', 
+    device='cuda',
     model_kwargs=None,
     text_channels=2048,
     apply_spatial_patchify=0,
@@ -295,6 +295,7 @@ def load_visual_tokenizer(args):
 
 def load_transformer(vae, args):
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    print(f"Device: {device}")
     model_path = args.model_path
     if args.checkpoint_type == 'torch': 
         # copy large model to local; save slim to local; and copy slim to nas; load local slim model
@@ -420,8 +421,8 @@ weights_path.mkdir(exist_ok=True)
 download_infinity_weights(weights_path)
 
 # Device setup
-device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 dtype = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.is_bf16_supported() else torch.float32
+print(f"Using dtype: {dtype}")
 
 # Define args
 args = argparse.Namespace(