mokady commited on
Commit
5cc5619
1 Parent(s): 9280b9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,7 +20,7 @@ pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
20
  pipe = pipe.to("cuda")
21
  pipe.force_zeros_for_empty_prompt = False
22
 
23
- print("Optimizing BRIA-LCM-2.2 - this could take a while")
24
  t=time.time()
25
  pipe.unet = torch.compile(
26
  pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
@@ -94,7 +94,7 @@ with gr.Blocks(css=css) as demo:
94
  resolution = gr.Dropdown(value=resolutions[0], show_label=True, label="Resolution", choices=resolutions)
95
  seed = gr.Textbox(label="Seed", value=-1)
96
  submit_btn = gr.Button("Generate")
97
- result = gr.Image(label="BRIA-LCM-2.2 Result")
98
 
99
  # gr.Examples(
100
  # examples = [
 
20
  pipe = pipe.to("cuda")
21
  pipe.force_zeros_for_empty_prompt = False
22
 
23
+ print("Optimizing BRIA 2.2 FAST - this could take a while")
24
  t=time.time()
25
  pipe.unet = torch.compile(
26
  pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
 
94
  resolution = gr.Dropdown(value=resolutions[0], show_label=True, label="Resolution", choices=resolutions)
95
  seed = gr.Textbox(label="Seed", value=-1)
96
  submit_btn = gr.Button("Generate")
97
+ result = gr.Image(label="BRIA 2.2 FAST Result")
98
 
99
  # gr.Examples(
100
  # examples = [