Shape Mismatch

#1
by pjain001 - opened

Getting the following error while running mflux 0.12.1:

mflux-generate --model exdysa/Flex.2-preview-MLX --prompt 'Test Prompt' --base-model schnell --steps 4 --seed 10 --width 1024 --height 1024

ValueError: [addmm] Last dimension of first input with shape (1,4096,64) must match second to last dimension of second input with shape (196,3072).

pjain001 changed discussion status to closed

I didnt try this one so out so I'm not all that surprised. Flex/OpenFlux has some unique qualities to it and will probably need some code adjustments on the mflux lib side. Ill look into forking the repo for it, thanks for giving it a shot anyway

Sign up or log in to comment