Open-Sora/tools/caption/acceleration/__init__.py
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

0 lines
Python