apple / ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner

Home Page:https://machinelearning.apple.com/research/matryoshka-diffusion-models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

web_ui 运行报错

fallbernana123456 opened this issue · comments

在 torchrun --standalone --nproc_per_node=1 ml_mdm/clis/generate_sample.py --port 运行后使用推荐的参数运行报错:

File "lib/python3.10/site-packages/ml_mdm/samplers.py", line 189, in read_gamma
return self.gammas[time]
RuntimeError: CUDA error: device-side assert triggered
Compile with TORCH_USE_CUDA_DSA to enable device-side assertions.