Update app.py

app.py

@@ -14,8 +14,8 @@ resolutions = ["1024 1024","1280 768","1344 768","768 1344","768 1280" ]
 # Load pipeline 
 
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
-unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.2-FAST", torch_dtype=torch.float16)
-pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.2", torch_dtype=torch.float16, unet=unet, vae=vae)
+unet = UNet2DConditionModel.from_pretrained("briaai/BRIA-2.3-FAST", torch_dtype=torch.float16)
+pipe = DiffusionPipeline.from_pretrained("briaai/BRIA-2.3-BETA", torch_dtype=torch.float16, unet=unet, vae=vae)
 pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
 pipe.to('cuda')
 del unet
@@ -24,26 +24,26 @@ del vae
 
 pipe.force_zeros_for_empty_prompt = False
 
-print("Optimizing BRIA 2.2 FAST - this could take a while")
-t=time.time()
-pipe.unet = torch.compile(
-    pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
-)
-with torch.no_grad():
-    outputs = pipe(
-        prompt="an apple",
-        num_inference_steps=8,
-    )
-
-    # This will avoid future compilations on different shapes
-    unet_compiled = torch._dynamo.run(pipe.unet)
-    unet_compiled.config=pipe.unet.config
-    unet_compiled.add_embedding = Dummy()
-    unet_compiled.add_embedding.linear_1 = Dummy()
-    unet_compiled.add_embedding.linear_1.in_features = pipe.unet.add_embedding.linear_1.in_features
-    pipe.unet = unet_compiled
-
-print(f"Optimizing finished successfully after {time.time()-t} secs")
+# print("Optimizing BRIA 2.3 FAST - this could take a while")
+# t=time.time()
+# pipe.unet = torch.compile(
+#     pipe.unet, mode="reduce-overhead", fullgraph=True # 600 secs compilation
+# )
+# with torch.no_grad():
+#     outputs = pipe(
+#         prompt="an apple",
+#         num_inference_steps=8,
+#     )
+
+#     # This will avoid future compilations on different shapes
+#     unet_compiled = torch._dynamo.run(pipe.unet)
+#     unet_compiled.config=pipe.unet.config
+#     unet_compiled.add_embedding = Dummy()
+#     unet_compiled.add_embedding.linear_1 = Dummy()
+#     unet_compiled.add_embedding.linear_1.in_features = pipe.unet.add_embedding.linear_1.in_features
+#     pipe.unet = unet_compiled
+
+# print(f"Optimizing finished successfully after {time.time()-t} secs")
 
 @spaces.GPU(enable_queue=True)
 def infer(prompt,seed,resolution):
@@ -84,16 +84,16 @@ css = """
 """
 with gr.Blocks(css=css) as demo:
     with gr.Column(elem_id="col-container"):
-        gr.Markdown("## BRIA 2.2 FAST")
+        gr.Markdown("## BRIA 2.3 FAST")
         gr.HTML('''
           <p style="margin-bottom: 10px; font-size: 94%">
             This is a demo for 
-            <a href="https://huggingface.co/briaai/BRIA-2.2-FAST" target="_blank">BRIA 2.2 FAST </a>. 
-            This is a fast version of BRIA 2.2 text-to-image model, still trained on licensed data, and so provides full legal liability coverage for copyright and privacy infringement.
-            Try it for free in our webapp demo <a href="https://labs.bria.ai/" </a>.
+            <a href="https://huggingface.co/briaai/BRIA-2.3-FAST" target="_blank">BRIA 2.3 FAST </a>. 
+            This is a fast version of BRIA 2.3 text-to-image model, still trained on licensed data, and so provides full legal liability coverage for copyright and privacy infringement.
+            You can also try it for free in our <a href="https://labs.bria.ai/" target="_blank">webapp demo </a>.
 
-            Are you a startup or a student? We encourage you to apply for our Startup Plan 
-            <a href="https://pages.bria.ai/the-visual-generative-ai-platform-for-builders-startups-plan?_gl=1*cqrl81*_ga*MTIxMDI2NzI5OC4xNjk5NTQ3MDAz*_ga_WRN60H46X4*MTcwOTM5OTMzNC4yNzguMC4xNzA5Mzk5MzM0LjYwLjAuMA..) </a> 
+            Are you a startup or a student? We encourage you to apply for our 
+            <a href="https://pages.bria.ai/the-visual-generative-ai-platform-for-builders-startups-plan?_gl=1*cqrl81*_ga*MTIxMDI2NzI5OC4xNjk5NTQ3MDAz*_ga_WRN60H46X4*MTcwOTM5OTMzNC4yNzguMC4xNzA5Mzk5MzM0LjYwLjAuMA..)  target="_blank">Startup Plan  </a>
             This program are designed to support emerging businesses and academic pursuits with our cutting-edge technology.
 
           </p>