huggingface / huggingface/diffusers
When I use deepspeed_zero3 to train the controlnet on flux, an error occured "RuntimeError: tensor does not have a device".
未关闭
stale
- 主要语言
- Python
- 星标
- 34.5k
- 派生
- 7.3k
- 平均合并
- 3 天 3 小时
- 30 天内合并 PR
- 91
描述
[rank0]: File "/home/z/.local/lib/python3.10/site-packages/deepspeed/ops/adam/cpu_adam.py", line 163, in step
[rank0]: self.ds_opt_adam.adam_update(self.opt_id, state['step'], group['lr'], beta1, beta2, group['eps'],
[rank0]: RuntimeError: tensor does not have a device
Steps: 0%| | 0/500000 [00:04
贡献指南
调研方向
Begin with the reported DeepSpeed ZeRO-3 ControlNet-on-FLUX training setup and inspect deepspeed/ops/adam/cpu_adam.py at line 163, where the failure occurs. Reproduce the error, determine which tensor lacks a device, and document a confirmed correction with the reproduction result.
由索引模型根据 Issue 内容生成。
评估
- 技术栈
- python, pytorch
- 领域
- machine-learning
- Issue 类型
- 缺陷
- 难度
- 4/5
- 预计耗时
- 3-5 天
- 活跃度
- 停滞
- 描述清晰度
- 需要澄清
- 新手友好度
- 25/100