Fix LTX2 channels-last review scope
Signed-off-by: Hongsheng Liu <liuhongsheng4@huawei.com>
This commit is contained in:
1 parent
81e5e8e3bb
commit
dc072a12bd
11 files changed
+35
-60
No files matched your search
@@ -534,18 +534,6 @@ class TestDiffusionCompileConfig:
|
||||
with pytest.raises(TypeError, match="diffusion_compile_dynamic"):
|
||||
OmniDiffusionConfig(model="test", diffusion_compile_dynamic="false")
|
||||
|
||||
def test_channels_last_3d_is_opt_in_and_strict(self) -> None:
|
||||
assert OmniDiffusionConfig(model="test").vae_use_channels_last_3d is False
|
||||
assert (
|
||||
OmniDiffusionConfig.from_kwargs(
|
||||
model="test",
|
||||
vae_use_channels_last_3d=True,
|
||||
).vae_use_channels_last_3d
|
||||
is True
|
||||
)
|
||||
with pytest.raises(TypeError, match="vae_use_channels_last_3d"):
|
||||
OmniDiffusionConfig(model="test", vae_use_channels_last_3d="true")
|
||||
|
||||
@pytest.mark.parametrize(
|
||||
"kwargs, feature",
|
||||
[
|
||||
|
||||
Reference in new issue
Block a user