avoid memory issue

app.py

@@ -21,7 +21,7 @@ else:
 
 pipelines = {
     model_id: DiffusionPipeline.from_pretrained(
-        model_id, device=device, dtype=torch_dtype
+        model_id, torch_dtype=torch_dtype
     )
     for model_id in model_ids
 }