Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -34,6 +34,7 @@ To optimize image results:
|
|
| 34 |
|
| 35 |
# Image Editor
|
| 36 |
pipe_edit = StableDiffusionInstructPix2PixPipeline.from_pretrained( "vinesmsuic/magicbrush-jul7", torch_dtype=torch.float16 ).to("cuda")
|
|
|
|
| 37 |
|
| 38 |
# Generator
|
| 39 |
@spaces.GPU(duration=30, queue=False)
|
|
|
|
| 34 |
|
| 35 |
# Image Editor
|
| 36 |
pipe_edit = StableDiffusionInstructPix2PixPipeline.from_pretrained( "vinesmsuic/magicbrush-jul7", torch_dtype=torch.float16 ).to("cuda")
|
| 37 |
+
pipe_edit.scheduler = EulerAncestralDiscreteScheduler.from_config(pipe_edit.scheduler.config)
|
| 38 |
|
| 39 |
# Generator
|
| 40 |
@spaces.GPU(duration=30, queue=False)
|