ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update sdxl supports_compile expectation after the U-Net compile change

Browse source 
The SDXL trainer now regionally compiles its transformer blocks, so /info
advertises supports_compile for every family; the precision selector stays
DiT-only.
This commit is contained in:
Daniel Han 2026-07-04 01:34:35 +00:00
commit c3b77c08d6
1 changed files with 3 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

4
studio/backend/tests/test_diffusion_base_precision.py
View file

@ -265,7 +265,9 @@ def test_family_train_infos_carries_precision_fields(monkeypatch):
sdxl = infos["sdxl"]
assert sdxl["precision_modes"] == []
assert sdxl["recommended_precision"] == "nf4"
assert sdxl["supports_compile"] is False
# The SDXL trainer regionally compiles its U-Net blocks too, so compile is advertised
# for every family; only the precision selector stays DiT-only.
assert sdxl["supports_compile"] is True
# ── request model base_precision field ────────────────────────────────────────