Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -61,7 +61,7 @@ if image is not None:
|
|
61 |
|
62 |
model_id = "timbrooks/instruct-pix2pix"
|
63 |
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
64 |
-
|
65 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
66 |
|
67 |
images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
|
|
|
61 |
|
62 |
model_id = "timbrooks/instruct-pix2pix"
|
63 |
pipe = StableDiffusionInstructPix2PixPipeline.from_pretrained(model_id, torch_dtype=torch.float16, safety_checker=None)
|
64 |
+
|
65 |
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe.scheduler.config)
|
66 |
|
67 |
images = pipe(prompt, image=image, num_inference_steps=10, image_guidance_scale=1).images
|