[Model] Add circlestone-labs/Anima (#4083)
Signed-off-by: akshatvishu <akshatnayak197@gmail.com>
This commit is contained in:
1 parent
698f716012
commit
415005be22
33 files changed
+3127
-46
No files matched your search
@@ -1,6 +1,8 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
|
||||
from tests.model_tests.diffusion.case_filtering import get_parametrized_options
|
||||
@@ -47,10 +49,17 @@ def test_pipeline_on_supported_tasks(
|
||||
# since starting the server can take 10+ seconds, even for tiny models.
|
||||
#
|
||||
# NOTE: Be sure to install pytest-subtests if you're running on pytest < 9
|
||||
settings = DIFFUSION_TEST_SETTINGS[model_name]
|
||||
model_path = tiny_model_paths[model_name]
|
||||
model_kwargs = {}
|
||||
if settings.checkpoint_filename is not None:
|
||||
model_path = str(Path(model_path) / settings.checkpoint_filename)
|
||||
model_kwargs["model_class_name"] = model_name
|
||||
omni = build_omni_from_diff_accelerations(
|
||||
accelerations=accelerations,
|
||||
model=tiny_model_paths[model_name],
|
||||
model=model_path,
|
||||
enforce_eager=True,
|
||||
**model_kwargs,
|
||||
)
|
||||
try:
|
||||
for task_type in supported_tasks:
|
||||
|
||||
Reference in new issue
Block a user