DDIM loop 内小张量分配优化,attention mask 缓存到 GPU

This commit is contained in:
qhy
2026-02-10 16:53:00 +08:00
parent ed637c972b
commit 91a9b0febc
5 changed files with 283 additions and 32 deletions

1
.gitignore vendored
View File

@@ -55,7 +55,6 @@ coverage.xml
*.pot
# Django stuff:
*.log
local_settings.py
db.sqlite3