Views
No views yet
llama-3.2-3b-finetune 中分别训练 Task1/2/3 后,用 PEFT 线性加权合并 得到的 单一 adapter(非全量合并后的独立权重文件;推理仍需加载下面的 基座模型)。meta-llama/Llama-3.2-3B-Instruct(需 Hugging Face 许可与 token)| 文件 | 说明 |
|---|---|
adapter_config.json / adapter_model.safetensors | 合并后的 LoRA |
tokenizer*.json, special_tokens_map.json, chat_template.jinja | 与训练时一致的 tokenizer |
merge_meta.json | 合并来源路径、权重(各任务 1/3 归一化)、combination_type: linear |
1import torch
2from transformers import AutoModelForCausalLM, AutoTokenizer
3from peft import PeftModel
4
5BASE = "meta-llama/Llama-3.2-3B-Instruct"
6ADAPTER = "/path/to/caiti-llama32-3b-ultimate" # 本目录
7
8tok_kw = {"token": "hf_..."} # 或已 huggingface-cli login
9
10base = AutoModelForCausalLM.from_pretrained(
11 BASE,
12 torch_dtype=torch.bfloat16,
13 device_map="auto",
14 **tok_kw,
15)
16model = PeftModel.from_pretrained(base, ADAPTER)
17tokenizer = AutoTokenizer.from_pretrained(ADAPTER, **tok_kw)llama-3.2-3b-finetune/llama_finetune_caiti.py(Task1 / Task2)llama-3.2-3b-finetune/llama32_3b_finetune_task3.py(Task3)llama-3.2-3b-finetune 下:1python3 merge_llama32_caiti_loras.py --out-dir output/merged_task123_linear
2# 再将 output/merged_task123_linear/merged/* 与 tokenizer 等拷贝到本目录(或改脚本直接 --out-dir 指向本路径)merge_llama32_caiti_loras.py → 覆盖本目录中的 adapter 与 merge_meta.json。CaiTI/llama-3.2-3b-finetune/llama-3.2-3b-finetune/README_MULTITASK.md