argmaxinc/DiffusionKit

Flux schnell hardcoded

Opened this issue · 1 comments

ngaloppo commented

This looks like a bug. Shouldn't the model config depend on the input model id?

DiffusionKit/python/src/diffusionkit/mlx/model_io.py

Line 706 in e63501d

config = FLUX_SCHNELL

arda-argmax commented

Thanks for letting us know. We will release a fix in the coming days.