Two evict/corruption fixes surfaced by review of the diffusion training path: - krea-2 sets force_bf16 in the DiT trainer spec, but the route-level _FORCE_BF16_FAMILIES preflight listed only qwen-image and z-image, so a krea-2 start with mixed_precision=fp16 passed the route check, reserved training and evicted resident GPU models, and only the child trainer then raised. Add krea-2 to the set and a drift-guard test asserting it equals the trainer specs whose force_bf16 is set. - The dataset-upload same-stem duplicate check compared stems case-sensitively, so on case-insensitive filesystems (Windows / default macOS) sample.png and Sample.jpg both passed even though their caption sidecars sample.txt / Sample.txt resolve to the same file, silently sharing and corrupting one caption. Compare stems and the same-name guard with casefold at both the on-disk and in-batch sites. |
||
|---|---|---|
| .. | ||
| data_recipe | ||
| export | ||
| inference | ||
| rag | ||
| training | ||
| __init__.py | ||
| _torchao_stub.py | ||
| import_guards.py | ||
| tool_healing.py | ||