#!/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 '  "'