instruct-pix2pix

Sleeping

salomonsky commited on Oct 19, 2024

Commit

74b12aa

verified ·

1 Parent(s): e2be69c

Update edit_app.py

Files changed (1) hide show

edit_app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from diffusers import StableDiffusionInstructPix2PixPipeline
 model_id = "timbrooks/instruct-pix2pix"
 def main():
-    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None).to("cpu")
     def generate(
         input_image: Image.Image,

 model_id = "timbrooks/instruct-pix2pix"
 def main():
+    pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, safety_checker=None).to("cpu")
     def generate(
         input_image: Image.Image,