update training yaml

salesforce · Jan 30, 2023 · ccffeca · ccffeca
1 parent bd7cac1
commit ccffeca
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/lavis/projects/blip2/train/caption_coco_ft.yaml b/lavis/projects/blip2/train/caption_coco_ft.yaml
@@ -4,8 +4,8 @@
  # For full license text, see the LICENSE file in the repo root or https://opensource.org/licenses/BSD-3-Clause
 
 model:
-  arch: blip_opt
-  model_type: caption_opt2.7b
+  arch: blip2_opt
+  model_type: caption_coco_opt2.7b
   load_finetuned: False
   use_grad_checkpoint: True
 

diff --git a/lavis/projects/blip2/train/pretrain.yaml b/lavis/projects/blip2/train/pretrain.yaml
@@ -5,7 +5,6 @@
 
 model:
   arch: blip2
-
   model_type: pretrain
   load_pretrained: False
-Original file line number
+Diff line change
@@ Expand Up / @@ -5,7 +5,6 @@ @@
     model:
       arch: blip2
       model_type: pretrain
       load_pretrained: False
@@ Expand Down @@