oahzxl commited on
Commit
7b637b3
1 Parent(s): 9c115c5

update slicing

Browse files
videosys/pipelines/cogvideox/pipeline_cogvideox.py CHANGED
@@ -101,6 +101,7 @@ class CogVideoXPipeline(VideoSysPipeline):
101
  vae = AutoencoderKLCogVideoX.from_pretrained(config.model_path, subfolder="vae", torch_dtype=self._dtype)
102
  if config.vae_tiling:
103
  vae.enable_tiling()
 
104
  if tokenizer is None:
105
  tokenizer = T5Tokenizer.from_pretrained(config.model_path, subfolder="tokenizer")
106
  if text_encoder is None:
 
101
  vae = AutoencoderKLCogVideoX.from_pretrained(config.model_path, subfolder="vae", torch_dtype=self._dtype)
102
  if config.vae_tiling:
103
  vae.enable_tiling()
104
+ vae.enable_slicing()
105
  if tokenizer is None:
106
  tokenizer = T5Tokenizer.from_pretrained(config.model_path, subfolder="tokenizer")
107
  if text_encoder is None: