saiboxx/chexray-diffusion

text2img model does not support gpu inference

Closed this issue · 2 comments

text2img model does not support gpu inference

The tokenized text embeddings were not properly transferred to the chosen device in the LDM base.
I fixed it with d6d3ffe 🔨 .

Can you please confirm that the error is resolved at your end?

The tokenized text embeddings were not properly transferred to the chosen device in the LDM base. I fixed it with d6d3ffe 🔨 .

Can you please confirm that the error is resolved at your end?

It works! Thank you!