nigeljw commited on
Commit
651e443
1 Parent(s): ef54619

Fixed bug with num inference steps slider and disabled progress bar

Browse files
Files changed (1) hide show
  1. app.py +4 -3
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