Open-Sora/opensora/models/layers
Zheng Zangwei (Alex Zheng) 45df92849c Merge pull request #156 from hpcaitech/feature/causal_atten
Added causal mask in Attention forward pass
2024-06-26 23:36:37 +08:00
..
__init__.py
blocks.py Merge pull request #156 from hpcaitech/feature/causal_atten 2024-06-26 23:36:37 +08:00