Update run_caption_ds.py

run_caption_ds.py

@@ -1,3 +1,9 @@
+'''
+git clone https://modelscope.cn/models/LLM-Research/Meta-Llama-3.1-8B
+
+python run_caption_ds.py "svjack/Genshin-Impact-Couple-with-Tags-IID-Gender-Only-Two" --caption_column="joy-caption" --output_path="gen_couple_cap_dir"
+'''
+
 import argparse
 from pathlib import Path
 import torch
@@ -9,7 +15,8 @@ from tqdm import tqdm  # 引入 tqdm 用于显示进度条
 # Constants
 CLIP_PATH = "google/siglip-so400m-patch14-384"
 VLM_PROMPT = "A descriptive caption for this image:\n"
-MODEL_PATH = "meta-llama/Meta-Llama-3.1-8B"
+#MODEL_PATH = "meta-llama/Meta-Llama-3.1-8B"
+MODEL_PATH = "Meta-Llama-3.1-8B"
 CHECKPOINT_PATH = Path("wpkklhc6")
 
 # Image Adapter