KingNish commited on
Commit
dde6a3a
Β·
verified Β·
1 Parent(s): ed1a78a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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)