31 lines
735 B
Python
31 lines
735 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
"""Reusable RL training primitives."""
|
|
|
|
from fastvideo.train.methods.rl.common.sampling import (
|
|
DiffusionSampler,
|
|
SamplingConfig,
|
|
SamplingResult,
|
|
)
|
|
from fastvideo.train.methods.rl.common.prompt_sampling import (
|
|
KRepeatSample,
|
|
distributed_k_repeat_indices,
|
|
)
|
|
from fastvideo.train.methods.rl.common.validation import (
|
|
RLValidationConfig,
|
|
media_to_video_array,
|
|
validation_caption,
|
|
validation_shard_indices,
|
|
)
|
|
|
|
__all__ = [
|
|
"DiffusionSampler",
|
|
"KRepeatSample",
|
|
"RLValidationConfig",
|
|
"SamplingConfig",
|
|
"SamplingResult",
|
|
"distributed_k_repeat_indices",
|
|
"media_to_video_array",
|
|
"validation_caption",
|
|
"validation_shard_indices",
|
|
]
|