6392bd40a9
Co-authored-by: SolitaryThinker <wlsaidhi@gmail.com> Co-authored-by: Matthew Noto <99706358+RandNMR73@users.noreply.github.com> Co-authored-by: XOR-op <17672363+XOR-op@users.noreply.github.com> Co-authored-by: Zhang Peiyuan <42993249+jzhang38@users.noreply.github.com>
110 lines
2.4 KiB
Python
110 lines
2.4 KiB
Python
# SPDX-License-Identifier: Apache-2.0
|
|
from fastvideo.api.schema import (
|
|
CompileConfig,
|
|
ComponentConfig,
|
|
ContinuationState,
|
|
EngineConfig,
|
|
GenerationPlan,
|
|
GenerationRequest,
|
|
GeneratorConfig,
|
|
GpuPoolConfig,
|
|
InputConfig,
|
|
OffloadConfig,
|
|
OutputConfig,
|
|
ParallelismConfig,
|
|
PipelineSelection,
|
|
PlannedStage,
|
|
PromptEnhancerConfig,
|
|
PromptSafetyConfig,
|
|
QuantizationConfig,
|
|
RequestRuntimeConfig,
|
|
RunConfig,
|
|
SamplingConfig,
|
|
ServeConfig,
|
|
ServerConfig,
|
|
StreamingConfig,
|
|
WarmupConfig,
|
|
)
|
|
from fastvideo.api.errors import ConfigValidationError
|
|
from fastvideo.api.overrides import apply_overrides, parse_cli_overrides
|
|
from fastvideo.api.presets import (
|
|
InferencePreset,
|
|
PresetStageSpec,
|
|
get_all_preset_names,
|
|
get_preset,
|
|
get_presets_for_family,
|
|
register_preset,
|
|
validate_preset_selection,
|
|
validate_stage_names,
|
|
validate_stage_overrides,
|
|
)
|
|
from fastvideo.api.parser import (
|
|
config_to_dict,
|
|
load_config,
|
|
load_raw_config,
|
|
load_run_config,
|
|
load_serve_config,
|
|
parse_config,
|
|
)
|
|
from fastvideo.api.results import (
|
|
GenerationResult,
|
|
VideoEvent,
|
|
VideoFinalEvent,
|
|
VideoPartialEvent,
|
|
VideoProgressEvent,
|
|
VideoResult,
|
|
)
|
|
from fastvideo.api.sampling_param import SamplingParam
|
|
|
|
__all__ = [
|
|
"CompileConfig",
|
|
"ComponentConfig",
|
|
"ContinuationState",
|
|
"ConfigValidationError",
|
|
"EngineConfig",
|
|
"GenerationResult",
|
|
"GenerationPlan",
|
|
"GenerationRequest",
|
|
"GeneratorConfig",
|
|
"GpuPoolConfig",
|
|
"InputConfig",
|
|
"OffloadConfig",
|
|
"OutputConfig",
|
|
"ParallelismConfig",
|
|
"PipelineSelection",
|
|
"PlannedStage",
|
|
"PromptEnhancerConfig",
|
|
"PromptSafetyConfig",
|
|
"QuantizationConfig",
|
|
"RequestRuntimeConfig",
|
|
"RunConfig",
|
|
"SamplingConfig",
|
|
"SamplingParam",
|
|
"ServeConfig",
|
|
"ServerConfig",
|
|
"StreamingConfig",
|
|
"VideoEvent",
|
|
"VideoFinalEvent",
|
|
"VideoPartialEvent",
|
|
"VideoProgressEvent",
|
|
"VideoResult",
|
|
"WarmupConfig",
|
|
"InferencePreset",
|
|
"PresetStageSpec",
|
|
"apply_overrides",
|
|
"config_to_dict",
|
|
"load_config",
|
|
"load_raw_config",
|
|
"load_run_config",
|
|
"load_serve_config",
|
|
"parse_cli_overrides",
|
|
"get_all_preset_names",
|
|
"get_preset",
|
|
"get_presets_for_family",
|
|
"parse_config",
|
|
"register_preset",
|
|
"validate_preset_selection",
|
|
"validate_stage_names",
|
|
"validate_stage_overrides",
|
|
]
|