You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/lazy
Edenzzzz f5c84af0b0
[Feature] Zigzag Ring attention (#5905)
3 months ago
..
__init__.py [misc] update pre-commit and run all files (#4752) 1 year ago
construction.py [lazy] support torch 2.0 (#4763) 1 year ago
lazy_init.py [doc] add lazy init docs (#4808) 1 year ago
pretrained.py [Feature] Zigzag Ring attention (#5905) 3 months ago