generation failed on AMD directml

ljleb / sd-webui-freeu

a1111 implementation of https://github.com/ChenyangSi/FreeU

MIT License

301 stars 16 forks source link

generation failed on AMD directml #21

Closed ananosleep closed 11 months ago

ananosleep commented 11 months ago

enable freeu and click generate, then:

0%| | 0/20 [00:00<?, ?it/s] [F D:\a_work\1\s\pytorch-directml-plugin\torch_directml\csrc\engine\dml_util.cc:118] Invalid or unsupported data type. press any key to continue...

the whole Stable Diffusion process exits directly.

using: https://github.com/lshqqytiger/stable-diffusion-webui-directml commit: 7b0b721

ljleb commented 11 months ago

Yes, it is a known issue. I was looking for how to solve this and I just found the solution. Will push a fix shortly.

ljleb commented 11 months ago

dml should do the fft on cpu now. Note that this will result in an expected slower generation time when the extension is enabled. see #16