Update README.md

README.md

@@ -6,9 +6,9 @@ language:
 pipeline_tag: image-text-to-text
 ---
 
-# probe_seg_llava-1.5-pt
+# probe_seg_llava-1.5-pt-ift
 
-This model checkpoint contains the seg probes for CLIP-ConvNeXT-XXL Llama-3-8b based LLaVA-1.5 model after the PT stage, i.e., trained on the LLaVA-558K dataset. Please refer to [documentation](https://github.com/SHI-Labs/OLA-VLM/blob/main/docs/Probing.md) for more details.
+This model checkpoint contains the seg probes for CLIP-ConvNeXT-XXL Llama-3-8b based LLaVA-1.5 model after the PT and IFT stages, i.e., trained on the LLaVA-558K and LLaVA-665K datasets. Please refer to [documentation](https://github.com/SHI-Labs/OLA-VLM/blob/main/docs/Probing.md) for more details.
 
 - **GitHub Repo:** [https://github.com/SHI-Labs/OLA-VLM](https://github.com/SHI-Labs/OLA-VLM)
 - **Project Page:** [https://praeclarumjj3.github.io/ola_vlm/](https://praeclarumjj3.github.io/ola_vlm/)