Spaces:
Running
Running
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:
|