Add MLX base-model DiT weights for LoRA training

#2
by cortexelus - opened
Stability AI org

dit_{sm-music,sm-sfx,medium}-base_f16.npz β€” the *-base (rectified_flow) DiT + baked conditioner in MLX layout, for LoRA training via optimized/mlx lora_train_mlx.py (registered in weights.TRAINING_BASE for lazy download).

mattricesound changed pull request status to merged

Sign up or log in to comment