Update README.md

README.md

@@ -124,13 +124,7 @@ mask_image = mask_image.convert("L")
 
 width, height = init_image.size
 
-# Load, init model
-controlnet = ControlNetModel().from_config('briaai/DEV-ControlNetInpaintingFast', torch_dtype=torch.float16)          
-controlnet.controlnet_cond_embedding = ControlNetConditioningEmbedding(
-    conditioning_embedding_channels=320,
-    conditioning_channels = 5
-)   
-    
+# Load, init model    
 controlnet = ControlNetModel().from_pretrained("briaai/DEV-ControlNetInpaintingFast", torch_dtype=torch.float16)
 vae = AutoencoderKL.from_pretrained("madebyollin/sdxl-vae-fp16-fix", torch_dtype=torch.float16)
 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)
@@ -144,7 +138,6 @@ pipe.enable_xformers_memory_efficient_attention()
 
 generator = torch.Generator(device='cuda:0').manual_seed(123456)
 
-
 vae = pipe.vae
 
 masked_image, image_mask, masked_image_to_present = get_masked_image(init_image, mask_image, width, height)