AutoModelForCausalLM 解析，因果模型
AutoModelForCausalLM 解析
目录
-
- AutoModelForCausalLM 解析
-
- 参数解析
-
- 1. `model_path`
- 2. `device_map='auto'`
- 其他常见参数
-
- 1. `cache_dir`
- 2. `force_download`
- 3. `resume_download`
- 4. `local_files_only`
- 5. `revision`
- 除了AutoModelForCausalLM ，还有什么：AutoModelForQuestionAnswering
-
- 1. 序列分类任务
- 2. 问答任务：AutoModelForQuestionAnswering
- 3. 标记任务（如命名实体识别）
- 4. 掩码语言模型任务：AutoModelForMaskedLM
- 5. 多模态任务：AutoModelForVision2Seq
- 6. 摘要任务
AutoModelForCausalLM.from_pretrained 是 Hugging Face Transformers 库中的一个方法，用于从预训练的模型权重加载一个因果语言模型。
参数解析
1. model_path
这是一个必需参数，代表预训练模型的路径或者模型名称。如果传入的是模型名称，模型会从 Hugging Face 的模型库中下载；若传入的是本地路径，模型会从该路径加载。
示例：
# 从 Hugging Face 模型库下载
model = AutoModelForCausalLM.from_pretrained("gpt2")
<