Update README.md

README.md

@@ -93,6 +93,14 @@ pipe.enable_offload(
     vae_offload=False,
 )
 
+# Optional: Use lighttae
+pipe.enable_lightvae(
+    use_tae=True,
+    tae_path="/path/to/lighttaehy1_5.safetensors",
+    use_lightvae=False,
+    vae_path=None,
+)
+
 # Create generator
 pipe.create_generator(
     attn_mode="sage_attn2",
@@ -142,6 +150,14 @@ pipe.enable_quantize(
     image_encoder_quantized=False,
 )
 
+# Optional: Use lighttae
+pipe.enable_lightvae(
+    use_tae=True,
+    tae_path="/path/to/lighttaehy1_5.safetensors",
+    use_lightvae=False,
+    vae_path=None,
+)
+
 # Optional: Enable offloading for lower VRAM usage
 pipe.enable_offload(
     cpu_offload=True,