看啥推荐读物
专栏名称: GiantPandaCV
专注于机器学习、深度学习、计算机视觉、图像处理等多个方向技术分享。团队由一群热爱技术且热衷于分享的小伙伴组成。我们坚持原创,每天一到两篇原创技术分享。希望在传播知识、分享知识的同时能够启发你,大家一起共同进步(・ω<)☆
今天看啥  ›  专栏  ›  GiantPandaCV

单节点8xA800跑起来HuggingFace DeepSeek V2踩坑

GiantPandaCV  · 公众号  · 3D  · 2024-05-09 15:55
0x0. 背景尝试跑起来HuggingFace上release的DeepSeek V2,踩了几个坑,这里给出解决的方法。HuggingFace提供的开源DeepSeek V2 repo链接为:https://huggingface.co/deepseek-ai/DeepSeek-V20x1. 报错1: KeyError: 'sdpa'这个问题社区也有人反馈了。https://huggingface.co/deepseek-ai/DeepSeek-V2/discussions/3在这里插入图片描述解决方法很简单,在工程里面的config.json最后加一句"_attn_implementation": "flash_attention_2"即可:在这里插入图片描述0x2. 报错2: 初始化阶段卡死已经给accelerate提了一个pr解决这个问题。https://github.com/huggingface/accelerate/pull/2756背景当我尝试使用 transformers 库进行 deepseek-v2 模型推理时:import torchfrom transformers import AutoTokenizer, AutoModelForCausalLM, GenerationConfigmodel_name = "deepseek-ai/DeepSeek-V2"tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)# `max_memory` should be set based on your de ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照