run on cpu

app.py

@@ -42,9 +42,9 @@ def set_pipeline(model_id_repo,scheduler):
 
     pipe = StableDiffusionPipeline.from_pretrained(
         model_repo,
-        torch_dtype=torch.float16,                               # to run on cpu
+        # torch_dtype=torch.float16,                               # to run on cpu
         use_safetensors=True,
-    ).to("cuda")
+    ).to("cpu")
 
 
     scheduler_classes = {