KingNish commited on
Commit
314fe06
Β·
verified Β·
1 Parent(s): 5683e67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -19,7 +19,6 @@ refiner = StableDiffusionXLImg2ImgPipeline.from_pretrained("stabilityai/stable-d
19
  refiner.to("cuda")
20
 
21
  pipe_fast = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V4.0_Lightning", torch_dtype=torch.float16)
22
- pipe_fast.scheduler = EulerAncestralDiscreteScheduler(pipe_fast.scheduler.config)
23
  pipe_fast.to("cuda")
24
 
25
  help_text = """
@@ -199,7 +198,7 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
199
  with gr.Row():
200
  type = gr.Dropdown(["Image Generation","Image Editing"], label="Task", value="Image Generation",interactive=True)
201
  enhance_prompt = gr.Checkbox(label="Enhance prompt", value=False, scale=0)
202
- fast = gr.Checkbox(label="FAST Generation", value=False, scale=0)
203
 
204
  with gr.Row():
205
  input_image = gr.Image(label="Image", type='filepath', interactive=True)
 
19
  refiner.to("cuda")
20
 
21
  pipe_fast = StableDiffusionXLPipeline.from_pretrained("SG161222/RealVisXL_V4.0_Lightning", torch_dtype=torch.float16)
 
22
  pipe_fast.to("cuda")
23
 
24
  help_text = """
 
198
  with gr.Row():
199
  type = gr.Dropdown(["Image Generation","Image Editing"], label="Task", value="Image Generation",interactive=True)
200
  enhance_prompt = gr.Checkbox(label="Enhance prompt", value=False, scale=0)
201
+ fast = gr.Checkbox(label="FAST Generation", value=True, scale=0)
202
 
203
  with gr.Row():
204
  input_image = gr.Image(label="Image", type='filepath', interactive=True)