Skip to content

Commit 7bb6dc9

Browse files
committed
ci(rvm): audit pinned MJ-VIDEO source import
1 parent 809552c commit 7bb6dc9

1 file changed

Lines changed: 24 additions & 0 deletions

File tree

.github/workflows/ci-rvm-mj-temporary.yml

Lines changed: 24 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,30 @@ jobs:
4646
examples/train/rvm_h3/preflight_mj_video.py \
4747
examples/train/rvm_h3/calibrate_reward_profile.py
4848
49+
- name: Import the exact pinned MJ-VIDEO source
50+
run: |
51+
git clone --filter=blob:none https://github.com/aiming-lab/MJ-Video.git /tmp/MJ-Video
52+
git -C /tmp/MJ-Video checkout --detach cc1d2c9587a620e9ebd3599ae4cdd21b5fd7c87a
53+
python - <<'PY'
54+
from pathlib import Path
55+
56+
from fastvideo.train.methods.rl.rewards.mj_video_compat import (
57+
install_mj_video_transformers_compat,
58+
)
59+
from fastvideo.train.methods.rl.rewards.mj_video import (
60+
_import_official_mj_video,
61+
)
62+
63+
install_mj_video_transformers_compat()
64+
model_cls, config_cls, prepare_chat_input = _import_official_mj_video(
65+
Path("/tmp/MJ-Video")
66+
)
67+
assert model_cls.__name__ == "InternVLChatRewardModeling"
68+
assert config_cls.__name__ == "InternVLChatRewardModelingConfig"
69+
assert callable(prepare_chat_input)
70+
print(model_cls, config_cls, prepare_chat_input)
71+
PY
72+
4973
- name: Parse shell and YAML artifacts
5074
run: |
5175
bash -n \

0 commit comments

Comments
 (0)