Update README.md

README.md

@@ -17,28 +17,34 @@ until the next Diffusers pypi release, please install Diffusers from source and
 TODO: change when new version.
 
 ## Control Mode
-| Control Mode | Description |
-|:------------:|:-----------:|
-|0|canny|
-|1|tile|
-|2|depth|
-|3|blur|
-|4|pose|
-|5|gray|
-|6|lq|
+| Control Mode | Description | Current Model Validity |
+|:------------:|:-----------:|:-----------:|
+|0|canny|🟢high|
+|1|tile|🟢high|
+|2|depth|🟡medium|
+|3|blur|🟢high|
+|4|pose|🔴low|
+|5|gray|🔴low|
+|6|lq|🟢high|
 
 ### canny
 <img src="./images/image_demo_canny.jpg" width = "800" />
+
 ### tile
 <img src="./images/image_demo_tile.jpg" width = "800" />
+
 ### depth
 <img src="./images/image_demo_depth.jpg" width = "800" />
+
 ### blur
 <img src="./images/image_demo_blur.jpg" width = "800" />
+
 ### pose
 <img src="./images/image_demo_pose.jpg" width = "800" />
+
 ### gray
 <img src="./images/image_demo_gray.jpg" width = "800" />
+
 ### low quality
 <img src="./images/image_demo_lq.jpg" width = "800" />
 
@@ -65,7 +71,7 @@ from diffusers.models.controlnet_flux import FluxControlNetModel
 
 # load
 base_model = 'black-forest-labs/FLUX.1-dev'
-controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Union-alpha '
+controlnet_model = 'InstantX/FLUX.1-dev-Controlnet-Union-alpha'
 controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
 pipe = FluxControlNetPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16)
 pipe.to("cuda")