bdfc77147d
Signed-off-by: Meng <meng_chen99@163.com> Signed-off-by: Yangshen Deng <yangshen.d@outlook.com> Signed-off-by: Suli Wang <50828270+wsl2000@users.noreply.github.com> Co-authored-by: Meng <meng_chen99@163.com> Co-authored-by: Suli Wang <50828270+wsl2000@users.noreply.github.com> Co-authored-by: amy-why-3459 <wuhaiyan17@huawei.com>
25 lines
730 B
Python
25 lines
730 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
import pytest
|
|
|
|
from vllm_omni.diffusion.models.dreamzero.utils import (
|
|
DEFAULT_CFG_SCALE,
|
|
DEFAULT_EMBODIMENT_NAME_TO_ID,
|
|
DEFAULT_NEGATIVE_PROMPT,
|
|
DEFAULT_NUM_INFERENCE_STEPS,
|
|
DEFAULT_SEED,
|
|
DEFAULT_SIGMA_SHIFT,
|
|
)
|
|
|
|
pytestmark = [pytest.mark.core_model, pytest.mark.cpu]
|
|
|
|
|
|
def test_dreamzero_default_constants_match_source_baseline():
|
|
assert DEFAULT_NUM_INFERENCE_STEPS == 16
|
|
assert DEFAULT_CFG_SCALE == 5.0
|
|
assert DEFAULT_SIGMA_SHIFT == 5.0
|
|
assert DEFAULT_SEED == 1140
|
|
assert "worst quality" in DEFAULT_NEGATIVE_PROMPT
|
|
assert DEFAULT_EMBODIMENT_NAME_TO_ID["oxe_droid"] == 17
|