ColossalAI icon indicating copy to clipboard operation
ColossalAI copied to clipboard

[BUG]: In the inference stage i got the problem

Open ML-GCN opened this issue 3 years ago • 4 comments

🐛 Describe the bug

Through training to get a ckpt file In the inference stage i got the problem image Can you help me look at it? or Give a tutorial on inference? i want through hugging face way to load model but have failed how can i thank you

Environment

the Environment the same provided I can train but inference failed

ML-GCN avatar Nov 12 '22 11:11 ML-GCN

I modified the load_model_from_config function in 'txt2img.py': image and solve the problem, you can try it.

jwliu-cc avatar Nov 13 '22 14:11 jwliu-cc

I modified the load_model_from_config function in 'txt2img.py': image and solve the problem, you can try it.

Thank you very much for your reply,I'll try it now

ML-GCN avatar Nov 14 '22 01:11 ML-GCN

in the ldm/models/diffusion/ddpm.py line 538 & 539, it comments the first stage and con stage init method, so we need to load it manually . image

WuZifan avatar Nov 17 '22 03:11 WuZifan

Thank you

ML-GCN avatar Nov 26 '22 09:11 ML-GCN

We have updated a lot. This issue was closed due to inactivity. Thanks.

binmakeswell avatar Apr 13 '23 10:04 binmakeswell