Explicitly add text_emb_layers to model_args.

kohjingyu · Oct 27, 2023 · 6b183ac · 6b183ac
1 parent 8216fea
commit 6b183ac
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/main.py b/main.py
@@ -240,6 +240,7 @@ def main_worker(gpu, ngpus_per_node, args):
   model_args = models.GILLArgs()
   model_args.opt_version = args.opt_version
   model_args.visual_encoder = args.visual_model
+  model_args.text_emb_layers = [-1]
   model_args.freeze_lm = True
   model_args.freeze_vm = True
   model_args.n_visual_tokens = args.n_visual_tokens