Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
212 |
-
BRIA
|
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():
|