apple / ml-mdm

Train high-quality text-to-image diffusion models in a data & compute efficient manner
https://machinelearning.apple.com/research/matryoshka-diffusion-models
MIT License
399 stars 21 forks source link

web_ui 运行报错 #11

Open fallbernana123456 opened 1 month ago

fallbernana123456 commented 1 month ago

在 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.