Gabiburt commited on
Commit
18e20eb
1 Parent(s): 42d99aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -80,7 +80,6 @@ def get_size(init_image):
80
  device = "cuda" if torch.cuda.is_available() else "cpu"
81
 
82
  # Load, init model
83
- # controlnet = ControlNetModel().from_pretrained("briaai/DEV-ControlNetInpaintingFast", torch_dtype=torch.float16)
84
  controlnet = ControlNetModel().from_pretrained("briaai/DEV-GenerativeFill", torch_dtype=torch.float16)
85
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
86
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained("briaai/BRIA-2.3", controlnet=controlnet.to(dtype=torch.float16), torch_dtype=torch.float16, vae=vae) #force_zeros_for_empty_prompt=False, # vae=vae)
@@ -208,8 +207,8 @@ with image_blocks as demo:
208
  gr.HTML('''
209
  <p style="margin-bottom: 10px; font-size: 94%">
210
  This is a demo for
211
- <a href="https://huggingface.co/briaai/DEV-GenerativeFill" target="_blank">BRIA Generative Fill </a>.
212
- BRIA Generative Fill improve the inpainting task for Modify/Add/Replace compared to BRIA 2.3 inpainting while still trained on licensed data, and so provide full legal liability coverage for copyright and privacy infringement.
213
  </p>
214
  ''')
215
  with gr.Row():
 
80
  device = "cuda" if torch.cuda.is_available() else "cpu"
81
 
82
  # Load, init model
 
83
  controlnet = ControlNetModel().from_pretrained("briaai/DEV-GenerativeFill", torch_dtype=torch.float16)
84
  vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
85
  pipe = StableDiffusionXLControlNetPipeline.from_pretrained("briaai/BRIA-2.3", controlnet=controlnet.to(dtype=torch.float16), torch_dtype=torch.float16, vae=vae) #force_zeros_for_empty_prompt=False, # vae=vae)
 
207
  gr.HTML('''
208
  <p style="margin-bottom: 10px; font-size: 94%">
209
  This is a demo for
210
+ <a href="https://huggingface.co/briaai/DEV-GenerativeFill" target="_blank">BRIA 2.3 ControlNet GenFill</a>.
211
+ BRIA GenFill focuses on the inpainting tasks Modify, Add, Replace and trained on licensed data, and provides full legal liability coverage for copyright and privacy infringement.
212
  </p>
213
  ''')
214
  with gr.Row():