update

app.py

@@ -22,15 +22,18 @@ else:
 MAX_SEED = np.iinfo(np.int32).max
 CACHE_EXAMPLES = os.getenv("CACHE_EXAMPLES", "1") == "1"
 MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "1024"))
-#TOKEN = os.getenv("HF_TOKEN")
+DEFAULT_IMAGE_SIZE = 1024
+torch_dtype = torch.float16
+if device == "cpu" or device == "mps":
+    DEFAULT_IMAGE_SIZE = 512
+    torch_dtype = torch.float32
 
 
 pipe_id = "SPRIGHT-T2I/spright-t2i-sd2"
 pipe = DiffusionPipeline.from_pretrained(
     pipe_id,
-    torch_dtype=torch.float16,
+    torch_dtype=torch_dtype,
     use_safetensors=True,
-#    token=TOKEN,
 ).to(device)
 
 
@@ -114,14 +117,14 @@ with gr.Blocks(css="style.css") as demo:
                 minimum=256,
                 maximum=MAX_IMAGE_SIZE,
                 step=32,
-                value=1024,
+                value=DEFAULT_IMAGE_SIZE,
             )
             height = gr.Slider(
                 label="Height",
                 minimum=256,
                 maximum=MAX_IMAGE_SIZE,
                 step=32,
-                value=1024,
+                value=DEFAULT_IMAGE_SIZE,
             )
         with gr.Row():
             guidance_scale = gr.Slider(