Update README.md

README.md

@@ -7,7 +7,7 @@ tags:
 pipeline_tag: visual-document-retrieval
 ---
 
-# ColSmolVLM-alpha: Visual Retriever based on SmolVLM-Instruct with ColBERT strategy
+# ColSmolVLM-v0.1: Visual Retriever based on SmolVLM-Instruct with ColBERT strategy
 
 ### This is a version trained with batch_size 128 for 3 epochs
 
@@ -57,12 +57,12 @@ from PIL import Image
 from colpali_engine.models import ColIdefics3, ColIdefics3Processor
 
 model = ColIdefics3.from_pretrained(
-        "vidore/colsmolvlm-alpha",
+        "vidore/colsmolvlm-v0.1",
         torch_dtype=torch.bfloat16,
         device_map="cuda:0",
         attn_implementation="flash_attention_2" # or eager
     ).eval()
-processor = ColIdefics3Processor.from_pretrained("vidore/colsmolvlm-alpha")
+processor = ColIdefics3Processor.from_pretrained("vidore/colsmolvlm-v0.1")
 
 # Your inputs
 images = [