fix memory

app.py

@@ -1,6 +1,7 @@
 import os
 
 os.environ["GRADIO_TEMP_DIR"] = os.path.join(os.getcwd(), ".tmp_outputs")
+os.environ["PYTORCH_CUDA_ALLOC_CONF"] = "expandable_segments:True"
 
 import logging
 import uuid

videosys/pipelines/cogvideox/pipeline_cogvideox.py

@@ -100,8 +100,7 @@ class CogVideoXPipeline(VideoSysPipeline):
         if vae is None:
             vae = AutoencoderKLCogVideoX.from_pretrained(config.model_path, subfolder="vae", torch_dtype=self._dtype)
             if config.vae_tiling:
-                vae.enable_tiling()
-                vae.enable_slicing()
+                vae.enable_tiling(tile_sample_min_height=vae.tile_sample_min_height // 2)
         if tokenizer is None:
             tokenizer = T5Tokenizer.from_pretrained(config.model_path, subfolder="tokenizer")
         if text_encoder is None: