[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
4e32ef6484
commit
b799b240ff
1 changed files with 3 additions and 1 deletions
|
|
@ -213,7 +213,9 @@ def test_family_train_infos_lists_dit_families(dit_train_host):
|
|||
assert "4bit" in infos["z-image"]["default_base"].lower()
|
||||
|
||||
|
||||
def test_family_train_infos_sdxl_supports_compile_without_precision_modes(monkeypatch, dit_train_host):
|
||||
def test_family_train_infos_sdxl_supports_compile_without_precision_modes(
|
||||
monkeypatch, dit_train_host
|
||||
):
|
||||
# Regional compile now applies to every family (the SDXL trainer compiles its U-Net blocks too),
|
||||
# but base_precision stays DiT-only, so SDXL advertises no precision modes while z-image keeps
|
||||
# its own. Pin the precision list so the assertion holds regardless of the host GPU.
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue