[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,8 +1,12 @@
|
||||
# SPDX-License-Identifier: Apache-2.0
|
||||
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
||||
|
||||
"""
|
||||
Utilities for resolving real models to their tiny model equivalents.
|
||||
"""
|
||||
|
||||
import logging
|
||||
from pathlib import Path
|
||||
|
||||
from tests.model_tests.diffusion.model_settings import DIFFUSION_TEST_SETTINGS
|
||||
from vllm_omni.diffusion.data import resolve_model_class_name
|
||||
@@ -30,4 +34,7 @@ def resolve_tiny_model_path(model: str) -> str:
|
||||
)
|
||||
return model
|
||||
|
||||
return test_opts.builder()
|
||||
model_path = test_opts.builder()
|
||||
if test_opts.checkpoint_filename is not None:
|
||||
return str(Path(model_path) / test_opts.checkpoint_filename)
|
||||
return model_path
|
||||
Reference in new issue
Block a user