Spaces:
Sleeping
Sleeping
yash
commited on
Commit
·
2cd7dfb
1
Parent(s):
40e9a12
remove scheduler
Browse files
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"
|
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)
|