32 lines
1.0 KiB
Python
32 lines
1.0 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""Request model for updating settings."""
|
|
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class SettingsUpdate(BaseModel):
|
|
defaultModelId: str | None = None
|
|
defaultModelIdT2v: str | None = None
|
|
defaultModelIdI2v: str | None = None
|
|
defaultModelIdT2i: str | None = None
|
|
numInferenceSteps: int | None = None
|
|
numFrames: int | None = None
|
|
height: int | None = None
|
|
width: int | None = None
|
|
guidanceScale: float | None = None
|
|
guidanceRescale: float | None = None
|
|
fps: int | None = None
|
|
seed: int | None = None
|
|
numGpus: int | None = None
|
|
ditCpuOffload: bool | None = None
|
|
textEncoderCpuOffload: bool | None = None
|
|
vaeCpuOffload: bool | None = None
|
|
imageEncoderCpuOffload: bool | None = None
|
|
useFsdpInference: bool | None = None
|
|
enableTorchCompile: bool | None = None
|
|
vsaSparsity: float | None = None
|
|
tpSize: int | None = None
|
|
spSize: int | None = None
|
|
autoStartJob: bool | None = None
|
|
datasetUploadPath: str | None = None
|