Make mochi-1 work

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