add from pretrained

This commit is contained in:
Prince Canuma
2026-01-23 18:13:51 +01:00
parent ce39e744c3
commit ef76ec0921
2 changed files with 30 additions and 19 deletions

View File

@@ -273,6 +273,7 @@ class VideoDecoderModelConfig(BaseModelConfig):
norm_type: Enum = None
causality_axis: Enum = None
dropout: float = 0.0
timestep_conditioning: bool = False
@dataclass
class VideoEncoderModelConfig(BaseModelConfig):