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