mindspore-lab / mindone

one for all, Optimal generator with No Exception
Apache License 2.0
328 stars 62 forks source link

ModuleNotFoundError: No module named 'sgm' #423

Open hw-liang opened 2 months ago

hw-liang commented 2 months ago

I was trying to run python svd_tools/convert.py under examples/svd/. To solve the gm import error, I followed instructions from https://github.com/mindspore-lab/mindone/issues/123.

export PYTHONPATH=/path to/mindone/examples/stable_diffusion_xl

But I met this error: TypeError: init() got an unexpected keyword argument 'load_first_stage_model'

where should I import gm for svd training?

hadipash commented 2 months ago

Thanks for reporting the bug! It has now been fixed in PR #430.