Hugging Face / 魔搭模型下载|缓存管理标准化实操指南
📌 先定核心原则:模型管理的 3 个铁律
🚀 标准化模型下载实操方案
一、代码直连下载(工程开发首选)
# 1. 切换Hugging Face国内镜像,速度提升10倍import osos.environ["HF_ENDPOINT"] = "https://hf-mirror.com"from transformers import AutoModelForCausalLM, AutoTokenizer# 2. 指定模型名称,自动下载并缓存至默认目录model_name = "qwen/Qwen2-7B-Instruct"tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
二、国内魔搭社区下载(镜像失效备选)
pip install modelscope
from modelscope import AutoModelForCausalLM, AutoTokenizer# 魔搭模型ID,和Hugging Face一一对应model_name = "qwen/Qwen2-7B-Instruct"tokenizer = AutoTokenizer.from_pretrained(model_name)model = AutoModelForCausalLM.from_pretrained(model_name)
三、手动下载(大模型 / 离线环境必备)
# 直接加载本地模型路径,完全离线可用local_model_path = "D:/AI-Models/Qwen2-7B-Instruct"tokenizer = AutoTokenizer.from_pretrained(local_model_path, trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained(local_model_path, trust_remote_code=True)
💾 本地缓存管理规范(告别磁盘爆炸)
1. 全局缓存目录自定义(一劳永逸)
from transformers import AutoModelForCausalLM, AutoTokenizermodel_name = "qwen/Qwen2-7B-Instruct"# 指定缓存目录,和全局配置不冲突tokenizer = AutoTokenizer.from_pretrained(model_name,cache_dir="D:/AI-Models/huggingface-cache",trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained(model_name,cache_dir="D:/AI-Models/huggingface-cache",trust_remote_code=True)
夜雨聆风