Signed-off-by: wangyu <410167048@qq.com>
This commit is contained in:
1 parent
1f0ed96fcd
commit
907f932b72
9 files changed
+60
-48
No files matched your search
@@ -716,10 +716,9 @@ def test_distilled_pipeline_initializes_sde_scheduler(
|
||||
stub_real_pipeline_init,
|
||||
monkeypatch: pytest.MonkeyPatch,
|
||||
) -> None:
|
||||
from diffusers import FlowMatchEulerDiscreteScheduler
|
||||
|
||||
from vllm_omni.diffusion.models.cosmos3.pipeline_cosmos3 import (
|
||||
Cosmos3OmniDiffusersPipeline,
|
||||
FlowMatchEulerDiscreteScheduler,
|
||||
)
|
||||
|
||||
t_list = [1.0, 0.9375, 0.8333333333333334, 0.625]
|
||||
|
||||
Reference in new issue
Block a user