19 lines
768 B
Bash
19 lines
768 B
Bash
#!/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 ' "' |