diff --git a/nodes.py b/nodes.py
index 59bfb2f..d210373 100644
--- a/nodes.py
+++ b/nodes.py
@@ -844,6 +844,7 @@ class TeaCache:
                         delattr(diffusion_model, 'accumulated_rel_l1_distance')
             
             current_percent = current_step_index / (len(sigmas) - 1)
+            c["transformer_options"]["current_percent"] = current_percent
             if start_percent <= current_percent <= end_percent:
                 c["transformer_options"]["enable_teacache"] = True
             else:
diff --git a/pyproject.toml b/pyproject.toml
index f916a33..a08d3ae 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "teacache"
 description = "Unofficial implementation of [ali-vilab/TeaCache](https://github.com/ali-vilab/TeaCache) for ComfyUI"
-version = "1.6.0"
+version = "1.6.1"
 license = {file = "LICENSE"}
 
 [project.urls]