Spaces:
Paused
Paused
Fixed bug with num inference steps slider and disabled progress bar
Browse files
app.py
CHANGED
@@ -50,6 +50,7 @@ def InitializeOutpainting():
|
|
50 |
#safety_checker=lambda images, **kwargs: (images, False))
|
51 |
|
52 |
outpaintPipeline.scheduler = DPMSolverMultistepScheduler.from_config(outpaintPipeline.scheduler.config)
|
|
|
53 |
|
54 |
# Based on: https://discuss.pytorch.org/t/help-regarding-slerp-function-for-generative-model-sampling/32475/4
|
55 |
# Further optimized to trade a divide operation for a multiply
|
@@ -121,9 +122,9 @@ with gradio.Blocks(live=True) as ux:
|
|
121 |
mask = gradio.Image(label="Mask", type="pil", value=defaultMask)
|
122 |
prompt = gradio.Textbox(label="Prompt", info=promptDesc, placeholder="A person in a room with colored hair", lines=3)
|
123 |
negativePrompt = gradio.Textbox(label="Negative Prompt", info=negPromptDesc, placeholder="Facial hair", lines=3)
|
124 |
-
guidanceScale = gradio.Slider(label="Guidance Scale", info="A higher value causes the generation to be more relative to the text prompt conditioning.", maximum=100, minimum=1, value=7.5)
|
125 |
-
numInferenceSteps = gradio.Slider(label="Number of Inference Steps", info=numInfStepsDesc, maximum=100, minimum=1, value=20)
|
126 |
-
generatorSeed = gradio.Slider(label="Generator Seed", info=generatorSeedDesc, maximum=10000, value=lastSeed)
|
127 |
#modelIndex = gradio.Dropdown(modelNames, label="Model", value="runwayml/stable-diffusion-inpainting")
|
128 |
inputImage.style(full_width=True)
|
129 |
|
|
|
50 |
#safety_checker=lambda images, **kwargs: (images, False))
|
51 |
|
52 |
outpaintPipeline.scheduler = DPMSolverMultistepScheduler.from_config(outpaintPipeline.scheduler.config)
|
53 |
+
outpaintPipeline.set_progress_bar_config(disable=True)
|
54 |
|
55 |
# Based on: https://discuss.pytorch.org/t/help-regarding-slerp-function-for-generative-model-sampling/32475/4
|
56 |
# Further optimized to trade a divide operation for a multiply
|
|
|
122 |
mask = gradio.Image(label="Mask", type="pil", value=defaultMask)
|
123 |
prompt = gradio.Textbox(label="Prompt", info=promptDesc, placeholder="A person in a room with colored hair", lines=3)
|
124 |
negativePrompt = gradio.Textbox(label="Negative Prompt", info=negPromptDesc, placeholder="Facial hair", lines=3)
|
125 |
+
guidanceScale = gradio.Slider(label="Guidance Scale", info="A higher value causes the generation to be more relative to the text prompt conditioning.", maximum=100, minimum=1, value=7.5, step= 0.1)
|
126 |
+
numInferenceSteps = gradio.Slider(label="Number of Inference Steps", info=numInfStepsDesc, maximum=100, minimum=1, value=20, step=1)
|
127 |
+
generatorSeed = gradio.Slider(label="Generator Seed", info=generatorSeedDesc, maximum=10000, minimum=1, value=lastSeed, step=1)
|
128 |
#modelIndex = gradio.Dropdown(modelNames, label="Model", value="runwayml/stable-diffusion-inpainting")
|
129 |
inputImage.style(full_width=True)
|
130 |
|