nkkbr commited on
Commit
a1a8e99
·
verified ·
1 Parent(s): 712d5ca

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -142,7 +142,7 @@ def load_video(video_path, max_frames_num,fps=1,force_sample=False):
142
  spare_frames = vr.get_batch(frame_idx).asnumpy()
143
  return spare_frames,frame_time,video_time
144
 
145
- pretrained = "nkkbr/ViCA2-stage2-onevision-ft"
146
  model_name = "vica_qwen"
147
  device = "cuda"
148
  device_map = "auto"
 
142
  spare_frames = vr.get_batch(frame_idx).asnumpy()
143
  return spare_frames,frame_time,video_time
144
 
145
+ pretrained = "nkkbr/ViCA2"
146
  model_name = "vica_qwen"
147
  device = "cuda"
148
  device_map = "auto"