[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
@@ -7,6 +7,8 @@ online serving stack (CLI arg parsing, subprocess, API routing, response
|
||||
encoding) using tiny models.
|
||||
"""
|
||||
|
||||
from pathlib import Path
|
||||
|
||||
import pytest
|
||||
from xdist import is_xdist_worker
|
||||
|
||||
@@ -68,6 +70,10 @@ def test_online_on_supported_tasks(
|
||||
model_path = tiny_model_paths[model_name]
|
||||
server_args = build_server_args_from_diff_accelerations(accelerations)
|
||||
server_args.append("--enforce-eager")
|
||||
settings = DIFFUSION_TEST_SETTINGS[model_name]
|
||||
if settings.checkpoint_filename is not None:
|
||||
model_path = str(Path(model_path) / settings.checkpoint_filename)
|
||||
server_args.extend(["--model-class-name", model_name])
|
||||
|
||||
with OmniServer(model_path, server_args) as server:
|
||||
# TODO: We may want to revisit run_level validation here,
|
||||
|
||||
Reference in new issue
Block a user