Update vicuna7b_v2.yaml

salesforce · Dec 1, 2023 · 27662ab · 27662ab
1 parent 522d9e8
commit 27662ab
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/projects/xinstructblip/demo/configs/vicuna7b_v2.yaml b/projects/xinstructblip/demo/configs/vicuna7b_v2.yaml
@@ -16,9 +16,9 @@ model:
   pc_model: "ulip2_pointbert"
   video_model: "eva_clip_g"
   audio_model: "beats"
-  pretrained_image_qformer: https://storage.googleapis.com/sfr-xinstructblip-data-research/model/xinstructblip_checkpoints/vicuna7b/image_qformer_improved.pth
+  pretrained_image_qformer: https://storage.googleapis.com/sfr-xinstructblip-data-research/model/xinstructblip_checkpoints/vicuna7b/image_qformer.pth
   pretrained_pc_qformer:  https://storage.googleapis.com/sfr-xinstructblip-data-research/model/xinstructblip_checkpoints/vicuna7b/pc_qformer_improved.pth
-  pretrained_video_qformer: https://storage.googleapis.com/sfr-xinstructblip-data-research/model/xinstructblip_checkpoints/vicuna7b/video_qformer_improved.pth
+  pretrained_video_qformer: https://storage.googleapis.com/sfr-xinstructblip-data-research/model/xinstructblip_checkpoints/vicuna7b/video_qformer.pth
   pretrained_audio_qformer: https://storage.googleapis.com/sfr-xinstructblip-data-research/model/xinstructblip_checkpoints/vicuna7b/audio_qformer_improved.pth
   load_attention_image_qformer: True
   load_attention_pc_qformer: True
@@ -75,4 +75,4 @@ model:
   prefix: "USER: "
   postfix: "\nASSISTANT:"
   predict_with_gen: False
-  clean_tokenization: True
+  clean_tokenization: True