Skip to content

Commit 383bb96

Browse files
committed
Update the recipe
Signed-off-by: Jingyu Xin <jingyux@nvidia.com>
1 parent fa0681f commit 383bb96

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

examples/diffusers/quantization/utils.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,9 @@ def filter_func_flux_dev(name: str) -> bool:
8383

8484
def filter_func_wan_video(name: str) -> bool:
8585
"""Filter function specifically for LTX-Video models."""
86-
pattern = re.compile(r".*(patch_embedding|condition_embedder|proj_out).*")
86+
pattern = re.compile(
87+
r".*(patch_embedding|condition_embedder|proj_out|blocks.0\.|blocks.1\.|blocks.39|blocks.38).*"
88+
)
8789
return pattern.match(name) is not None
8890

8991

0 commit comments

Comments
 (0)