Update README.md

README.md

@@ -133,8 +133,8 @@ except:
 hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='pipeline_bria.py', local_dir=local_dir)
 hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='transformer_bria.py', local_dir=local_dir)
 hf_hub_download(repo_id="briaai/BRIA-4B-Adapt", filename='bria_utils.py', local_dir=local_dir)
-hf_hub_download(repo_id="briaai/BRIA-4B-Adapt-ControlNet-Union", filename='pipeline_bria_controlnet.py', local_dir=local_dir)
-hf_hub_download(repo_id="briaai/BRIA-4B-Adapt-ControlNet-Union", filename='controlnet_bria.py', local_dir=local_dir)
+hf_hub_download(repo_id="briaai/BRIA-3.0-ControlNet-Union", filename='pipeline_bria_controlnet.py', local_dir=local_dir)
+hf_hub_download(repo_id="briaai/BRIA-3.0-ControlNet-Union", filename='controlnet_bria.py', local_dir=local_dir)
 
 
 import torch
@@ -144,7 +144,7 @@ from pipeline_bria_controlnet import BriaControlNetPipeline
 import PIL.Image as Image
 
 base_model = 'briaai/BRIA-4B-Adapt'
-controlnet_model = 'briaai/BRIA-4B-Adapt-ControlNet-Union'
+controlnet_model = 'briaai/BRIA-3.0-ControlNet-Union'
 
 controlnet = BriaControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 controlnet = BriaMultiControlNetModel([controlnet])
@@ -152,8 +152,8 @@ controlnet = BriaMultiControlNetModel([controlnet])
 pipe = BriaControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16, trust_remote_code=True)
 pipe.to("cuda")
 
-control_image_colorgrid = load_image("https://huggingface.co/briaai/BRIA-4B-Adapt-ControlNet-Union/resolve/main/images/colorgrid.jpg")
-control_image_pose = load_image("https://huggingface.co/briaai/BRIA-4B-Adapt-ControlNet-Union/resolve/main/images/pose.jpg")
+control_image_colorgrid = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/colorgrid.jpg")
+control_image_pose = load_image("https://huggingface.co/briaai/BRIA-3.0-ControlNet-Union/resolve/main/pose.jpg")
 
 control_image = [control_image_colorgrid, control_image_pose]
 controlnet_conditioning_scale = [0.5, 0.5]
@@ -161,7 +161,7 @@ control_mode = [2, 5]
 
 width, height = control_image[0].size
 
-prompt = 'Two kids in jackets play near a tent in a forest with trees.'
+prompt = 'Two kids in jackets play near a tent in a forest.'
 
 generator = torch.Generator(device="cuda").manual_seed(555)
 image = pipe(
@@ -174,7 +174,7 @@ image = pipe(
     num_inference_steps=50, 
     max_sequence_length=128,
     guidance_scale=5,
-    generator=generator
+    generator=generator,
+    negative_prompt="Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate"
 ).images[0]
-
 ```