nicolabs/Open-Sora
1
0
Fork 0
mirror of https://github.com/hpcaitech/Open-Sora.git synced 2026-05-03 12:24:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1
Open-Sora/tools/caption/acceleration/llava
History
Frank Lee b704d6c0f8 Feature/llava speedup (#2) 
* [caption] accelerated llava with flash attention and parallel frame extraction

* supported dp and tp in llava

* code formatting
2024-03-27 16:55:25 +08:00
..
__init__.py Feature/llava speedup (#2) 2024-03-27 16:55:25 +08:00
policy.py Feature/llava speedup (#2) 2024-03-27 16:55:25 +08:00