TIGER-Lab
/

VLM2Vec-Full

Text Generation

Model card Files Files and versions

wenhu commited on Oct 14, 2024

Commit

c326e8a

·

verified ·

1 Parent(s): 6a18731

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -54,7 +54,7 @@ import numpy as np
 model_args = ModelArguments(
     model_name='microsoft/Phi-3.5-vision-instruct',
-    pooling='eos',
     normalize=True,
     lora=True,
     checkpoint_path='TIGER-Lab/VLM2Vec-LoRA')

 model_args = ModelArguments(
     model_name='microsoft/Phi-3.5-vision-instruct',
+    pooling='last',
     normalize=True,
     lora=True,
     checkpoint_path='TIGER-Lab/VLM2Vec-LoRA')