15 lines
341 B
Python
15 lines
341 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM-Omni project
|
|
"""Request models for Omni server control routes."""
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class OmniSleepRequest(BaseModel):
|
|
stage_ids: list[int]
|
|
level: int = 2
|
|
|
|
|
|
class OmniWakeupRequest(BaseModel):
|
|
stage_ids: list[int]
|