Spaces:
Runtime error
Runtime error
yash
commited on
Commit
·
c0e616b
1
Parent(s):
fc17416
remove scheduler
Browse files
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"
|
| 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)
|