Update README.md

README.md

@@ -25,24 +25,31 @@ You can find **DialoguePrompts** here: https://huggingface.co/datasets/XD-MU/Dia
 ```bash
 import os
 from huggingface_hub import snapshot_download
-from swift.llm import PtEngine, RequestConfig, InferRequest
+
 os.environ['CUDA_VISIBLE_DEVICES'] = '0'
 
 model_name = "XD-MU/ScriptAgent"
 local_path = "./models/ScriptAgent"
 
+# 下载整个仓库的所有文件
+print("下载模型所有文件...")
 snapshot_download(
     repo_id=model_name,
     local_dir=local_path,
-    local_dir_use_symlinks=False, 
-    resume_download=True 
+    local_dir_use_symlinks=False,
+    resume_download=True
 )
 
+print(f"模型已完整下载到: {local_path}")
+
+# 使用 SWIFT 加载
+from swift.llm import PtEngine, RequestConfig, InferRequest
+
 engine = PtEngine(local_path, max_batch_size=1)
 request_config = RequestConfig(max_tokens=8192, temperature=0.7)
 
 infer_request = InferRequest(messages=[
-    {"role": "user", "content": "Your Dialogue"}
+    {"role": "user", "content": "你的对话上下文(Your Dialogue)"}
 ])
 response = engine.infer([infer_request], request_config)[0]