search

argmaxinc / DiffusionKit

On-device Inference of Diffusion Models for Apple Silicon
MIT License
495 stars 30 forks source link

Flux schnell hardcoded #37

Open ngaloppo opened 1 month ago

ngaloppo commented 1 month ago

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

https://github.com/argmaxinc/DiffusionKit/blob/e63501d95e0aeafc1c31220a7262c6c74bdfc351/python/src/diffusionkit/mlx/model_io.py#L706

arda-argmax commented 1 month ago

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