From 1a61fe3cec2a4243a6776be27460e21c637270a2 Mon Sep 17 00:00:00 2001
From: YunjieYu <yjyu007@163.com>
Date: Fri, 7 Mar 2025 11:34:14 +0800
Subject: [PATCH] fix TeaCacheForImgGen node bug

---
 nodes.py       | 1 +
 pyproject.toml | 4 ++--
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index a7207f8..c9458a5 100644
--- a/nodes.py
+++ b/nodes.py
@@ -580,6 +580,7 @@ class TeaCacheForImgGen:
         if 'transformer_options' not in new_model.model_options:
             new_model.model_options['transformer_options'] = {}
         new_model.model_options["transformer_options"]["rel_l1_thresh"] = rel_l1_thresh
+        new_model.model_options["transformer_options"]["coefficients"] = SUPPORTED_MODELS_COEFFICIENTS[model_type]
         diffusion_model = new_model.get_model_object("diffusion_model")
 
         if model_type == "flux":
diff --git a/pyproject.toml b/pyproject.toml
index b048754..0946ffb 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
-name = "teacache"
+name = "ComfyUI-TeaCache"
 description = "Unofficial implementation of [ali-vilab/TeaCache](https://github.com/ali-vilab/TeaCache) for ComfyUI"
-version = "1.3.0"
+version = "1.3.1"
 license = {file = "LICENSE"}
 
 [project.urls]