File size: 654 Bytes
6ef31de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
device: 'cuda'
vicuna:
model_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID/vicuna-7b'
device: 'cuda'
num_gpus: 1
max_gpu_memory: '24Gib'
load_8bit: True
conv_template:
temperature: 1.0
max_new_tokens: 512
debug: False
output_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID/examples/vicuna_out.json'
vid2seq:
enable: True
clip_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID/clip_ckpt/ViT-L-14.pt'
output_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID/examples/'
work_dir: 'vid2seq_workdir'
config_path: 'config/vid2seq_config.py'
checkpoint_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID/vid2seq_ckpt' #only folder name
|