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