求助,跑ModelZoo中LLaMA 7B模型,报错507033 和E30003
收藏回复举报
求助,跑ModelZoo中LLaMA 7B模型,报错507033 和E30003
t('forum.solved') 已解决
发表于2024-01-12 15:27:54
0 查看

求助,跑ModelZoo中LLaMA 7B模型(https://www.hiascend.com/software/modelzoo/models/detail/296c23cf975c47b9b7a829b9586a5b0c),报错507033,我的机器是4张910B的昇腾卡32G,因为单卡内存不足加载完模型,做了流水线模型并行,4张卡加载一个模型,使用例子中的以下命令:

python ./tools/ckpt_convert/llama/convert_weights_from_huggingface.py \    --input-model-dir ./dataset/llama-7b-hf/ \    --output-model-dir ./dataset/llama-7b-as/ \    --tensor-model-parallel-size 1 \    --pipeline-model-parallel-size 4 \    --type 7B \    --deepspeed

执行shell脚本也做了对应修改:

deepspeed pretrain_llama.py \ --DDP-impl local \ --tensor-model-parallel-size 1 \ --pipeline-model-parallel-size 4 \ --num-layers 32 \ --hidden-size 4096 \

但结果报错507033 和E30003,求助怎么解决

%E6%90%9C%E7%8B%97%E6%88%AA%E5%9B%BE20240112152617.png

本帖最后由 匿名用户2024/01/12 18:11:03 编辑

我要发帖子