添加模型文件分片 (Epoch 7, 3 parts + merge script)

This commit is contained in:
shuiyu1123 committed 2026-06-16 12:10:24 +00:00
1 parent 91db488ad4
commit 9cef1d9ba1
5 files changed
+31

No files matched your search

+19
View File
@@ -0,0 +1,19 @@
#!/bin/bash
# 合并分片模型文件
# 用法: bash merge_model.sh
PARTS="llm_model_epoch7_part_aa llm_model_epoch7_part_ab llm_model_epoch7_part_ac llm_model_epoch7_part_ad"
OUTPUT="llm_model_v2_resume_epoch_7.pth"
echo "合并模型分片 → $OUTPUT ..."
cat $PARTS > $OUTPUT
echo "✅ 合并完成: $(ls -lh $OUTPUT | awk '{print $5}')"
echo ""
echo "使用方法:"
echo " python3 -c \"import json, torch;"
echo " from llm_chat_improved import LLMModel, LLMChat, tokenizer_instance;"
echo ' vocab = json.load(open("vocab_v2.json"));'
echo ' ckpt = torch.load("llm_model_v2_resume_epoch_7.pth", map_location="cpu");'
echo ' chat = LLMChat(LLMModel(...), vocab, device="cpu");'
echo ' print(chat.generate_response("小狸你好呀"));'
echo ' "'