CogVideo icon indicating copy to clipboard operation
CogVideo copied to clipboard

Published 20 hours ago • verified publisher icon THUDM

KeyError: 'shadow'

Open SpringtoString opened this issue 5 months ago • 4 comments

System Info / 系統信息

linux

Information / 问题信息

  • [X] The official example scripts / 官方的示例脚本
  • [ ] My own modified scripts / 我自己修改的脚本和任务

Reproduction / 复现过程

full-parameter fine-tuning on 2 h800 gpu

torchrun --standalone --nproc_per_node=8 train_video.py --base configs/cogvideox_2b.yaml configs/sft.yaml --seed $RANDOM

image

Expected behavior / 期待表现

solve KeyError: 'shadow' problem

SpringtoString avatar Aug 27 '24 08:08 SpringtoString