From 17bc41245abe84744619c282ca6b771f45569fdf Mon Sep 17 00:00:00 2001
From: yangmeng <yhyang1@gmail.com>
Date: Tue, 28 Jan 2025 23:03:05 +0800
Subject: [PATCH] fix: image generating in demos

---
 demo/app.py          | 4 ++--
 demo/app_januspro.py | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/demo/app.py b/demo/app.py
index 9edfc4f..bc6e32d 100644
--- a/demo/app.py
+++ b/demo/app.py
@@ -9,9 +9,9 @@ import numpy as np
 # Device and dtype configuration
 def get_device_and_dtype():
     if torch.cuda.is_available():
-        return 'cuda', torch.bfloat16
+        return 'cuda', torch.float32
     elif torch.backends.mps.is_available():
-        return 'mps', torch.float16
+        return 'mps', torch.float32
     return 'cpu', torch.float32
 
 device, dtype = get_device_and_dtype()
diff --git a/demo/app_januspro.py b/demo/app_januspro.py
index 25c9f2c..1e6c65f 100644
--- a/demo/app_januspro.py
+++ b/demo/app_januspro.py
@@ -15,10 +15,10 @@ if torch.cuda.is_available():
     dtype = torch.bfloat16
 elif torch.backends.mps.is_available():
     device = 'mps'
-    dtype = torch.float16
+    dtype = torch.float32  # MPS设备使用float32
 else:
     device = 'cpu'
-    dtype = torch.float16
+    dtype = torch.float32  # CPU设备使用float32
 
 # Load model and processor
 model_path = "deepseek-ai/Janus-Pro-7B"