Update README.md

README.md

@@ -91,13 +91,13 @@ pip install -r requirements.txt
 Download the necessary pre-trained models before proceeding to inference.
 
 ```shell
-STAR/checkpoints/STAR-7B.pt
+STAR/checkpoints/STAR-3B.pt
 STAR/checkpoints/VQ-Model.pt
 ```
 
 ### Configuration
 
-The model configuration file `star/configs/STAR_Qwen2.5-VL-7B.json` contains all necessary parameters for model initialization. Make sure to update the paths in the configuration file to match your local setup.
+The model configuration file `star/configs/STAR_Qwen2.5-VL-3B.json` contains all necessary parameters for model initialization. Make sure to update the paths in the configuration file to match your local setup.
 
 ## 🔥 Quick Start
 
@@ -120,7 +120,7 @@ python3 inference_understand.py \
     --image-path "path/to/your/image.jpg" \
     --question "What is in this image? Describe it in detail." \
     --max-new-tokens 256 \
-    --model-config "star/configs/STAR_Qwen2.5-VL-7B.json" \
+    --model-config "star/configs/STAR_Qwen2.5-VL-3B.json" \
     --checkpoint "checkpoints/STAR-7B.pt" \
     --device "cuda:0"
 ```
@@ -145,7 +145,7 @@ python3 inference_generation.py \
     --cfg 1.1 \
     --topk 1000 \
     --topp 0.8 \
-    --model-config "star/configs/STAR_Qwen2.5-VL-7B.json" \
+    --model-config "star/configs/STAR_Qwen2.5-VL-3B.json" \
     --checkpoint "checkpoints/STAR-7B.pt" \
     --diffusion-as-decoder \
     --device "cuda:0"
@@ -172,7 +172,7 @@ python3 inference_edit.py \
     --cfg 1.1 \
     --topk 1000 \
     --topp 0.8 \
-    --model-config "star/configs/STAR_Qwen2.5-VL-7B.json" \
+    --model-config "star/configs/STAR_Qwen2.5-VL-3B.json" \
     --checkpoint "checkpoints/STAR-7B.pt" \
     --diffusion-as-decoder \
     --device "cuda:0"