briaai
/

BRIA-4B-Adapt-ControlNet-Union

Model card Files Files and versions

TalHach61 commited on Feb 20

Commit

262dd81

·

verified ·

1 Parent(s): e27f711

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -99,7 +99,7 @@ def resize_img(control_image):
 base_model = 'briaai/BRIA-4B-Adapt'
 controlnet_model = 'briaai/BRIA-4B-Adapt-ControlNet-Union'
 controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
-pipeline = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
 pipeline = pipeline.to(device="cuda", dtype=torch.bfloat16)
 control_image_canny = load_image("https://huggingface.co/briaai/BRIA-4B-Adapt-ControlNet-Union/resolve/main/images/canny.jpg")

 base_model = 'briaai/BRIA-4B-Adapt'
 controlnet_model = 'briaai/BRIA-4B-Adapt-ControlNet-Union'
 controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
+pipeline = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, trust_remote_code=True)
 pipeline = pipeline.to(device="cuda", dtype=torch.bfloat16)
 control_image_canny = load_image("https://huggingface.co/briaai/BRIA-4B-Adapt-ControlNet-Union/resolve/main/images/canny.jpg")