Merge pull request #340 from mit-han-lab/dev

feat: support PuLID, Double FBCache and TeaCache; better linter

Merge pull request #340 from mit-han-lab/dev
feat: support PuLID, Double FBCache and TeaCache; better linter
37a27712 · Muyang Li · GitHub · c1d6fc84 · 760ab022 · 37a27712
Unverified Commit 37a27712 authored May 01, 2025 by Muyang Li Committed by GitHub May 01, 2025
20 changed files
--- a/scripts/build_docker.sh
+++ b/scripts/build_docker.sh
--- a/scripts/build_docker_torch27.sh
+++ b/scripts/build_docker_torch27.sh
--- a/scripts/build_docker_torch28.sh
+++ b/scripts/build_docker_torch28.sh
--- a/scripts/build_linux_wheel.sh
+++ b/scripts/build_linux_wheel.sh
--- a/scripts/build_linux_wheel_cu128.sh
+++ b/scripts/build_linux_wheel_cu128.sh
--- a/scripts/build_linux_wheel_torch2.7_cu128.sh
+++ b/scripts/build_linux_wheel_torch2.7_cu128.sh
--- a/scripts/linux_cleanup.sh
+++ b/scripts/linux_cleanup.sh
--- a/setup.py
+++ b/setup.py
--- a/src/FluxModel.cpp
+++ b/src/FluxModel.cpp
--- a/src/FluxModel.h
+++ b/src/FluxModel.h
--- a/src/Linear.cpp
+++ b/src/Linear.cpp
--- a/src/Linear.h
+++ b/src/Linear.h
--- a/src/Module.cpp
+++ b/src/Module.cpp
--- a/src/Module.h
+++ b/src/Module.h
--- a/src/SanaModel.cpp
+++ b/src/SanaModel.cpp
--- a/src/SanaModel.h
+++ b/src/SanaModel.h
--- a/src/Serialization.cpp
+++ b/src/Serialization.cpp
--- a/src/Serialization.h
+++ b/src/Serialization.h
--- a/src/Tensor.h
+++ b/src/Tensor.h
--- a/src/activation.cpp
+++ b/src/activation.cpp