Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ class Dummy():
|
|
12 |
pass
|
13 |
|
14 |
|
15 |
-
pipeline_path = snapshot_download(repo_id='briaai/BRIA-2.
|
16 |
sys.path.append(pipeline_path)
|
17 |
from ella_xl_pipeline import EllaXLPipeline
|
18 |
|
@@ -37,7 +37,7 @@ def tocuda():
|
|
37 |
pipe.pipe.unet.ella.to('cuda')
|
38 |
|
39 |
|
40 |
-
# print("Optimizing BRIA-2.
|
41 |
# t=time.time()
|
42 |
# pipe.unet = torch.compile(
|
43 |
# pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
|
@@ -106,12 +106,11 @@ css = """
|
|
106 |
"""
|
107 |
with gr.Blocks(css=css) as demo:
|
108 |
with gr.Column(elem_id="col-container"):
|
109 |
-
gr.Markdown("## BRIA 2.
|
110 |
gr.HTML('''
|
111 |
<p style="margin-bottom: 10px; font-size: 94%">
|
112 |
This is a demo for
|
113 |
-
<a href="https://huggingface.co/briaai/BRIA-2.
|
114 |
-
BRIA 2.4 improve the generation of humans and illustrations compared to BRIA 2.2 while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
|
115 |
</p>
|
116 |
''')
|
117 |
with gr.Group():
|
@@ -122,7 +121,7 @@ with gr.Blocks(css=css) as demo:
|
|
122 |
steps = gr.Textbox(label="Steps", value=50)
|
123 |
negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
|
124 |
submit_btn = gr.Button("Generate")
|
125 |
-
result = gr.Image(label="BRIA-2.
|
126 |
|
127 |
# gr.Examples(
|
128 |
# examples = [
|
|
|
12 |
pass
|
13 |
|
14 |
|
15 |
+
pipeline_path = snapshot_download(repo_id='briaai/BRIA-2.3-T5')
|
16 |
sys.path.append(pipeline_path)
|
17 |
from ella_xl_pipeline import EllaXLPipeline
|
18 |
|
|
|
37 |
pipe.pipe.unet.ella.to('cuda')
|
38 |
|
39 |
|
40 |
+
# print("Optimizing BRIA-2.3-T5 - this could take a while")
|
41 |
# t=time.time()
|
42 |
# pipe.unet = torch.compile(
|
43 |
# pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
|
|
|
106 |
"""
|
107 |
with gr.Blocks(css=css) as demo:
|
108 |
with gr.Column(elem_id="col-container"):
|
109 |
+
gr.Markdown("## BRIA 2.3 T5")
|
110 |
gr.HTML('''
|
111 |
<p style="margin-bottom: 10px; font-size: 94%">
|
112 |
This is a demo for
|
113 |
+
<a href="https://huggingface.co/briaai/BRIA-2.3-T5" target="_blank">BRIA 2.3 T5 text-to-image </a>.
|
|
|
114 |
</p>
|
115 |
''')
|
116 |
with gr.Group():
|
|
|
121 |
steps = gr.Textbox(label="Steps", value=50)
|
122 |
negative_prompt = gr.Textbox(label="Negative Prompt", value=default_negative_prompt)
|
123 |
submit_btn = gr.Button("Generate")
|
124 |
+
result = gr.Image(label="BRIA-2.3-T5 Result")
|
125 |
|
126 |
# gr.Examples(
|
127 |
# examples = [
|