fixed type for cpu

app.py

@@ -40,9 +40,11 @@ object_addition_finetuned_model_id = "paint-by-inpaint/add-finetuned-mb"
 general_editing_finetuned_model_id = "paint-by-inpaint/general-finetuned-mb"
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
+dtype = torch.float16 if "cuda" in device else torch.float32
+
 
 def load_model(model_id):
-    return StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to(device)
+    return StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=dtype).to(device)
 
 pipe_object_addition_base = load_model(object_addition_base_model_id)
 pipe_object_addition_finetuned = load_model(object_addition_finetuned_model_id)