From c3b77c08d6560ba684fac8a996dae5dc9133bcd9 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sat, 4 Jul 2026 01:34:35 +0000 Subject: [PATCH] Update sdxl supports_compile expectation after the U-Net compile change The SDXL trainer now regionally compiles its transformer blocks, so /info advertises supports_compile for every family; the precision selector stays DiT-only. --- studio/backend/tests/test_diffusion_base_precision.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/studio/backend/tests/test_diffusion_base_precision.py b/studio/backend/tests/test_diffusion_base_precision.py index 1f8ca9318f..4c27efcc60 100644 --- a/studio/backend/tests/test_diffusion_base_precision.py +++ b/studio/backend/tests/test_diffusion_base_precision.py @@ -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 ────────────────────────────────────────