assert len(mod.functions) == 1 error

[keilsmart]

File web_stable_diffusion/trace/model_trace.py", line 98, in vae_to_image
assert len(mod.functions) == 1
AssertionError

my enviroment is mlc-ai-nightly-0.12.dev1246, pytorch2.0.0.

Since I can't use pip3 install mlc-ai-nightly -f https://mlc.ai/wheels to install mlc-ai directory, so I download mlc-ai from https://github.com/mlc-ai/package/releases/tag/v0.9.dev0.

Is this caused by the mlc-ai version?

[Civitasv]

It should be 2.

[luismiaresse]

The same happened to me. You have to use the nightly version of pytorch (it is in their walkthrough notebook). Stable version gives that error