Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280" ]
|
|
13 |
|
14 |
# Load pipeline
|
15 |
|
16 |
-
unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-
|
17 |
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", unet=unet, torch_dtype=torch.float16)
|
18 |
del unet
|
19 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
@@ -80,12 +80,12 @@ css = """
|
|
80 |
"""
|
81 |
with gr.Blocks(css=css) as demo:
|
82 |
with gr.Column(elem_id="col-container"):
|
83 |
-
gr.Markdown("## BRIA 2.2
|
84 |
gr.HTML('''
|
85 |
<p style="margin-bottom: 10px; font-size: 94%">
|
86 |
This is a demo for
|
87 |
-
<a href="https://huggingface.co/briaai/BRIA-
|
88 |
-
|
89 |
</p>
|
90 |
''')
|
91 |
with gr.Group():
|
|
|
13 |
|
14 |
# Load pipeline
|
15 |
|
16 |
+
unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.2-FAST", torch_dtype=torch.float16)
|
17 |
pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", unet=unet, torch_dtype=torch.float16)
|
18 |
del unet
|
19 |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
|
|
|
80 |
"""
|
81 |
with gr.Blocks(css=css) as demo:
|
82 |
with gr.Column(elem_id="col-container"):
|
83 |
+
gr.Markdown("## BRIA 2.2 FAST")
|
84 |
gr.HTML('''
|
85 |
<p style="margin-bottom: 10px; font-size: 94%">
|
86 |
This is a demo for
|
87 |
+
<a href="https://huggingface.co/briaai/BRIA-2.2-FAST" target="_blank">BRIA 2.2 FAST </a>.
|
88 |
+
This is a fast version of BRIA 2.2 text-to-image model, still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
|
89 |
</p>
|
90 |
''')
|
91 |
with gr.Group():
|