yash commited on
Commit
2cd7dfb
·
1 Parent(s): 40e9a12

remove scheduler

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -3,7 +3,6 @@ import gradio as gr
3
  from diffusers import StableDiffusionPipeline
4
  from diffusers import ControlNetModel, DDIMScheduler,EulerDiscreteScheduler,EulerAncestralDiscreteScheduler,UniPCMultistepScheduler
5
  from diffusers import KDPM2DiscreteScheduler,KDPM2AncestralDiscreteScheduler,PNDMScheduler,StableDiffusionPipeline
6
- from diffusers import DPMSolverMultistepScheduler
7
  import random
8
 
9
  def set_pipeline(model_id_repo,scheduler):
@@ -32,8 +31,6 @@ def set_pipeline(model_id_repo,scheduler):
32
  "DPM2 Karras": KDPM2DiscreteScheduler,
33
  "DPM2 a Karras": KDPM2AncestralDiscreteScheduler,
34
  "PNDM": PNDMScheduler,
35
- "DPM++ 2M Karras": DPMSolverMultistepScheduler,
36
- "DPM++ 2M SDE Karras": DPMSolverMultistepScheduler,
37
  }
38
 
39
  sampler_name = scheduler # Example sampler name, replace with the actual value
@@ -98,7 +95,7 @@ with block as image_gen:
98
  run_btn = gr.Button("image generation", elem_id="select_btn", variant="primary")
99
  with gr.Accordion(label="Advance Options",open=False):
100
  model_selection = gr.Dropdown(choices=["runwayml","Realistic_Vision_V5_1_noVAE"],value="Realistic_Vision_V5_1_noVAE",label="Models")
101
- schduler_selection = gr.Dropdown(choices=["DDIM","Euler","Euler a","UniPC","DPM2 Karras","DPM2 a Karras","PNDM","DPM++ 2M Karras","DPM++ 2M SDE Karras"],value="Euler a",label="Scheduler")
102
  guidance_scale_slider = gr.Slider(label="guidance_scale", minimum=0, maximum=15, value=7.5, step=0.5)
103
  num_images_per_prompt_slider = gr.Slider(label="num_images_per_prompt", minimum=0, maximum=5, value=1, step=1)
104
  height_slider = gr.Slider(label="height", minimum=0, maximum=2048, value=896, step=1)
 
3
  from diffusers import StableDiffusionPipeline
4
  from diffusers import ControlNetModel, DDIMScheduler,EulerDiscreteScheduler,EulerAncestralDiscreteScheduler,UniPCMultistepScheduler
5
  from diffusers import KDPM2DiscreteScheduler,KDPM2AncestralDiscreteScheduler,PNDMScheduler,StableDiffusionPipeline
 
6
  import random
7
 
8
  def set_pipeline(model_id_repo,scheduler):
 
31
  "DPM2 Karras": KDPM2DiscreteScheduler,
32
  "DPM2 a Karras": KDPM2AncestralDiscreteScheduler,
33
  "PNDM": PNDMScheduler,
 
 
34
  }
35
 
36
  sampler_name = scheduler # Example sampler name, replace with the actual value
 
95
  run_btn = gr.Button("image generation", elem_id="select_btn", variant="primary")
96
  with gr.Accordion(label="Advance Options",open=False):
97
  model_selection = gr.Dropdown(choices=["runwayml","Realistic_Vision_V5_1_noVAE"],value="Realistic_Vision_V5_1_noVAE",label="Models")
98
+ schduler_selection = gr.Dropdown(choices=["DDIM","Euler","Euler a","UniPC","DPM2 Karras","DPM2 a Karras","PNDM"],value="Euler a",label="Scheduler")
99
  guidance_scale_slider = gr.Slider(label="guidance_scale", minimum=0, maximum=15, value=7.5, step=0.5)
100
  num_images_per_prompt_slider = gr.Slider(label="num_images_per_prompt", minimum=0, maximum=5, value=1, step=1)
101
  height_slider = gr.Slider(label="height", minimum=0, maximum=2048, value=896, step=1)