diff --git a/fastvideo/pipelines/preprocess/hunyuan/hunyuan_preprocess_pipelins.py b/fastvideo/pipelines/preprocess/hunyuan/hunyuan_preprocess_pipelins.py index fcf0ac4a9..9eb987210 100644 --- a/fastvideo/pipelines/preprocess/hunyuan/hunyuan_preprocess_pipelins.py +++ b/fastvideo/pipelines/preprocess/hunyuan/hunyuan_preprocess_pipelins.py @@ -5,58 +5,6 @@ from fastvideo.pipelines.preprocess.preprocess_stages import ( from fastvideo.pipelines.stages import (EncodingStage, ImageEncodingStage, TextEncodingStage) from fastvideo.pipelines.stages.image_encoding import ImageVAEEncodingStage - - -class PreprocessPipelineI2V(ComposedPipelineBase): - _required_config_modules = [ - "text_encoder", "tokenizer", "text_encoder_2", "tokenizer_2", "vae" - ] - - def create_pipeline_stages(self, fastvideo_args: FastVideoArgs): - assert fastvideo_args.preprocess_config is not None - self.add_stage(stage_name="text_transform_stage", - stage=TextTransformStage( - cfg_uncondition_drop_rate=fastvideo_args. - preprocess_config.training_cfg_rate, - seed=fastvideo_args.preprocess_config.seed, - )) - text_encoders = [ - self.get_module("text_encoder"), - self.get_module("text_encoder_2") - ] - tokenizers = [ - self.get_module("tokenizer"), - self.get_module("tokenizer_2") - ] - - self.add_stage(stage_name="prompt_encoding_stage", - stage=TextEncodingStage( - text_encoders=text_encoders, - tokenizers=tokenizers, - )) - self.add_stage( - stage_name="video_transform_stage", - stage=VideoTransformStage( - train_fps=fastvideo_args.preprocess_config.train_fps, - num_frames=fastvideo_args.preprocess_config.num_frames, - max_height=fastvideo_args.preprocess_config.max_height, - max_width=fastvideo_args.preprocess_config.max_width, - do_temporal_sample=fastvideo_args.preprocess_config. - do_temporal_sample, - )) - if (self.get_module("image_encoder") is not None - and self.get_module("image_processor") is not None): - self.add_stage( - stage_name="image_encoding_stage", - stage=ImageEncodingStage( - image_encoder=self.get_module("image_encoder"), - image_processor=self.get_module("image_processor"), - )) - self.add_stage(stage_name="image_vae_encoding_stage", - stage=ImageVAEEncodingStage( - vae=self.get_module("vae"), )) - self.add_stage(stage_name="video_encoding_stage", - stage=EncodingStage(vae=self.get_module("vae"), )) class PreprocessPipelineT2V(ComposedPipelineBase): @@ -108,7 +56,4 @@ class PreprocessPipelineT2V(ComposedPipelineBase): stage=EncodingStage(vae=self.get_module("vae")) ) -EntryClass = [PreprocessPipelineI2V, PreprocessPipelineT2V] - - - +EntryClass = [PreprocessPipelineT2V]