Update PhotoMaker v2 demo

app.py

@@ -42,7 +42,11 @@ DEFAULT_ASPECT_RATIO = ASPECT_RATIO_LABELS[0]
 enable_doodle_arg = False
 photomaker_ckpt = hf_hub_download(repo_id="TencentARC/PhotoMaker", filename="photomaker-v2.bin", repo_type="model")
 
-torch_dtype = torch.bfloat16 if torch.cuda.is_bf16_supported() else torch.float16
+if torch.cuda.is_available() and torch.cuda.is_bf16_supported():
+ torch_dtype = torch.bfloat16
+else:
+ torch_dtype = torch.float16
+ 
 if device == "mps":
  torch_dtype = torch.float16