Make mochi-1 work

I’ve tried to use the quantized imnotednamode/mochi-1-preview-mix-nf4-small
There might be an issue with the example. I think the mochi branch as merged, so this does not work:
pip install git+https://github.com/huggingface/diffusers@mochi
But the other issue, I don’t know how to resolve is that the MochiPileeline is looking for: model_index.json but It’s not available
HTTPError: 404 Client Error: Not Found for url: https://huggingface.co/genmo/mochi-1-preview/resolve/main/model_index.json

Any hint?

2 Likes

imnotednamode has answered. Mochi branch has been merged as I’ve suspected.
For the model load revision="refs/pr/18" has to be used:

pipe = MochiPipeline.from_pretrained("genmo/mochi-1-preview", revision="refs/pr/18", torch_dtype=torch.float16, transformer=transformer)
1 Like

There is a PR, but it hasn’t been merged into the main branch. Maybe they just forgot about it.

1 Like

Thanks,
Seems they’ve just merged it

1 Like

Oh, Great! :grinning:

This topic was automatically closed 12 hours after the last reply. New replies are no longer allowed.