update ui

cogvideox/ui/ui.py

@@ -27,7 +27,7 @@ from transformers import (CLIPImageProcessor, CLIPVisionModelWithProjection,
                           T5EncoderModel, T5Tokenizer)
 
 from cogvideox.data.bucket_sampler import ASPECT_RATIO_512, get_closest_ratio
-from ..models.autoencoder_magvit import AutoencoderKLCogVideoX
+from cogvideox.models.autoencoder_magvit import AutoencoderKLCogVideoX
 from cogvideox.models.transformer3d import CogVideoXTransformer3DModel
 from cogvideox.pipeline.pipeline_cogvideox import CogVideoX_Fun_Pipeline
 from cogvideox.pipeline.pipeline_cogvideox_inpaint import \
@@ -1282,7 +1282,7 @@ def ui_eas(model_name, savedir_sample):
                             show_label=False,
                             visible=True,
                         )
-                        length_slider = gr.Slider(label="Animation length (视频帧数)", value=49, minimum=5,   maximum=49,  step=4)
+                        length_slider = gr.Slider(label="Animation length (视频帧数)", value=25, minimum=5,   maximum=25,  step=4)
                     
                     source_method = gr.Radio(
                         ["Text to Video (文本到视频)", "Image to Video (图片到视频)", "Video to Video (视频到视频)"],
@@ -1347,7 +1347,7 @@ def ui_eas(model_name, savedir_sample):
 
             def upload_generation_method(generation_method):
                 if generation_method == "Video Generation":
-                    return gr.update(visible=True, minimum=5, maximum=49, value=49, interactive=True)
+                    return gr.update(visible=True, minimum=5, maximum=25, value=25, interactive=True)
                 elif generation_method == "Image Generation":
                     return gr.update(minimum=1, maximum=1, value=1, interactive=False)
             generation_method.change(