Update app.py

app.py

@@ -4,11 +4,10 @@ import torch
 
 def load_model():
     try:
-        # Load the model with explicit variant for half-precision weights
+        # Load the model without forcing half-precision
         pipeline = StableDiffusionPipeline.from_pretrained(
             "stabilityai/stable-diffusion-2-1",
-            torch_dtype=torch.float16,
-            variant="fp16",  # Updated from 'revision' to 'variant'
+            torch_dtype=torch.float32,  # Use float32 for CPU compatibility
             safety_checker=None  # Disable safety checker for faster inference
         )
     except Exception as e:
@@ -18,15 +17,16 @@ def load_model():
     # Configure the scheduler for faster generation
     pipeline.scheduler = DPMSolverMultistepScheduler.from_config(pipeline.scheduler.config)
 
-    # Move to CPU if no GPU is available
+    # Move to CPU
     try:
-        pipeline = pipeline.to("cuda" if torch.cuda.is_available() else "cpu")
+        pipeline = pipeline.to("cpu")
     except Exception as e:
         print(f"Error moving the model to device: {e}")
         raise
 
     return pipeline
 
+
 # Initialize the model
 try:
     model = load_model()