yash commited on
Commit
c0e616b
1 Parent(s): fc17416

remove scheduler

Browse files
Files changed (2) hide show
  1. README.md +2 -0
  2. app.py +1 -4
README.md CHANGED
@@ -10,3 +10,5 @@ pinned: false
10
  ---
11
 
12
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
 
 
 
10
  ---
11
 
12
  Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
13
+
14
+ Use prompt from Prompt.txt file
app.py CHANGED
@@ -3,7 +3,6 @@ import gradio as gr
3
  from diffusers import StableDiffusionPipeline
4
  from diffusers import DDIMScheduler,EulerDiscreteScheduler,EulerAncestralDiscreteScheduler,UniPCMultistepScheduler
5
  from diffusers import KDPM2DiscreteScheduler,KDPM2AncestralDiscreteScheduler,PNDMScheduler,StableDiffusionPipeline
6
- from diffusers import DPMSolverMultistepScheduler
7
  import random
8
 
9
 
@@ -39,8 +38,6 @@ def set_pipeline(model_id_repo,scheduler):
39
  "DPM2 Karras": KDPM2DiscreteScheduler,
40
  "DPM2 a Karras": KDPM2AncestralDiscreteScheduler,
41
  "PNDM": PNDMScheduler,
42
- "DPM++ 2M Karras": DPMSolverMultistepScheduler,
43
- "DPM++ 2M SDE Karras": DPMSolverMultistepScheduler,
44
  }
45
 
46
  sampler_name = scheduler # Example sampler name, replace with the actual value
@@ -106,7 +103,7 @@ with block as image_gen:
106
  run_btn = gr.Button("image generation", elem_id="select_btn", variant="primary")
107
  with gr.Accordion(label="Advance Options",open=False):
108
  model_selection = gr.Dropdown(choices=["pokemon","pokemon_v1.1"],value="pokemon",label="Models")
109
- 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")
110
  guidance_scale_slider = gr.Slider(label="guidance_scale", minimum=0, maximum=15, value=7.5, step=0.5)
111
  num_images_per_prompt_slider = gr.Slider(label="num_images_per_prompt", minimum=0, maximum=5, value=1, step=1)
112
  height_slider = gr.Slider(label="height", minimum=0, maximum=1024, value=512, step=1)
 
3
  from diffusers import StableDiffusionPipeline
4
  from diffusers import DDIMScheduler,EulerDiscreteScheduler,EulerAncestralDiscreteScheduler,UniPCMultistepScheduler
5
  from diffusers import KDPM2DiscreteScheduler,KDPM2AncestralDiscreteScheduler,PNDMScheduler,StableDiffusionPipeline
 
6
  import random
7
 
8
 
 
38
  "DPM2 Karras": KDPM2DiscreteScheduler,
39
  "DPM2 a Karras": KDPM2AncestralDiscreteScheduler,
40
  "PNDM": PNDMScheduler,
 
 
41
  }
42
 
43
  sampler_name = scheduler # Example sampler name, replace with the actual value
 
103
  run_btn = gr.Button("image generation", elem_id="select_btn", variant="primary")
104
  with gr.Accordion(label="Advance Options",open=False):
105
  model_selection = gr.Dropdown(choices=["pokemon","pokemon_v1.1"],value="pokemon",label="Models")
106
+ schduler_selection = gr.Dropdown(choices=["DDIM","Euler","Euler a","UniPC","DPM2 Karras","DPM2 a Karras","PNDM"],value="Euler a",label="Scheduler")
107
  guidance_scale_slider = gr.Slider(label="guidance_scale", minimum=0, maximum=15, value=7.5, step=0.5)
108
  num_images_per_prompt_slider = gr.Slider(label="num_images_per_prompt", minimum=0, maximum=5, value=1, step=1)
109
  height_slider = gr.Slider(label="height", minimum=0, maximum=1024, value=512, step=1)