File size: 460 Bytes
6ef31de ea7140b 6ef31de 49b3986 6ef31de ea7140b 6ef31de |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
device: 'cuda'
vicuna:
model_path: '/home/user/app/vicuna-7b'
# model_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID_huggingface/vicuna-7b'
device: 'cuda'
num_gpus: 'auto'
max_gpu_memory: '24Gib'
load_8bit: True
conv_template:
temperature: 1.0
max_new_tokens: 512
debug: False
output_path: '/home/user/app/vicuna_out.json'
# output_path: '/mnt/petrelfs/wangyiqin/vid_cap/ChatVID_huggingface/vicuna_out.json'
vid2seq:
enable: False
|