Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
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
|
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 = [
|