Commit Graph
4 Commits
Author SHA1 Message Date
hkz 58b27cee1c fix caption rewrite (#169) 2024-12-12 18:21:58 +08:00
hkz 485e771acf Fix train reward lora and video caption (#165) 2024-12-10 14:23:33 +08:00
hkz 30682ba3c5 Update video caption (#129) 2024-11-13 17:43:21 +08:00
hkz 5ea1bf2450 Update Video Caption (#93)
* Add VILA1.5 in video caption

* update VILA1.5 in video_caption

* add get_video_path_list back & update the accelerate version & auto load the vila quant_path & download models in the main process

* add exception handling & add force_ascii=False

* fix the NCCL timeout & use logger

* update utils

* Update video splitting

* Update pre-filtering

* Update video caption

* Add caption_rewrite.py

* Add VideoCLIPXL

* Add beautiful prompt demo

* fix test

* Update Dockerfile.ds

* update VideoCLIPXL & add filter_meta_train.py

* update stage3

* update README.md

* update README_zh-CN.md

* update README

* update requirements

* fix stage_3_video_recaptioning.sh

* update Beautiful Prompt

* update README

* update vila_video_recaptioning.py & fix the empty gather result
2024-08-21 16:40:54 +08:00