mokady commited on
Commit
9280b9e
·
verified ·
1 Parent(s): 068bcb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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-LCM-2.2", 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,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 LCM Beta")
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-LCM-2.2" target="_blank">BRIA LCM 2.2 text-to-image </a>.
88
- BRIA LCM 2.2 is a fast LCM version of BRIA 2.2 while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
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():