init

app.py

@@ -21,7 +21,7 @@ else:
     torch_dtype = torch.float32
     device = "cpu"
 
-pipelines = {model_id: DiffusionPipeline(model_id, device=device, dtype=torch_dtype) for model_id in model_ids}
+pipelines = {model_id: DiffusionPipeline.from_pretrained(model_id, device=device, dtype=torch_dtype) for model_id in model_ids}
 
 @spaces.GPU()
 def inference(