Spaces:

LanguageBind
/

Video-LLaVA

Running on A10G

LanguageBind commited on Feb 18

Commit

ccd1b1b

•

1 Parent(s): cbfb9b8

Update llava/model/builder.py

Files changed (1) hide show

llava/model/builder.py CHANGED Viewed

@@ -149,7 +149,7 @@ def load_pretrained_model(model_path, model_base, model_name, load_8bit=False, l
                 tokenizer.add_tokens([DEFAULT_X_START_TOKEN[x.upper()], DEFAULT_X_END_TOKEN[x.upper()]], special_tokens=True)
         '''
         model.resize_token_embeddings(len(tokenizer))
-        print(X)
         #if 'Image' in X:
         image_tower = model.get_image_tower()
         if not image_tower.is_loaded:

                 tokenizer.add_tokens([DEFAULT_X_START_TOKEN[x.upper()], DEFAULT_X_END_TOKEN[x.upper()]], special_tokens=True)
         '''
         model.resize_token_embeddings(len(tokenizer))
+        #print(X)
         #if 'Image' in X:
         image_tower = model.get_image_tower()
         if not image_tower.is_loaded: