From 96b2f17b1a0bb90dffd95b46378995ddd9d23b84 Mon Sep 17 00:00:00 2001 From: Aryan Date: Sun, 15 Sep 2024 20:27:44 +0200 Subject: [PATCH] fix --- examples/cogvideo/train_cogvideox_lora.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/examples/cogvideo/train_cogvideox_lora.py b/examples/cogvideo/train_cogvideox_lora.py index 93b90f50b9..55ea27d2f8 100644 --- a/examples/cogvideo/train_cogvideox_lora.py +++ b/examples/cogvideo/train_cogvideox_lora.py @@ -1541,6 +1541,11 @@ def main(args): ) pipe.scheduler = CogVideoXDPMScheduler.from_config(pipe.scheduler.config) + if args.enable_slicing: + pipe.vae.enable_slicing() + if args.enable_tiling: + pipe.vae.enable_tiling() + # Load LoRA weights lora_scaling = args.lora_alpha / args.rank pipe.load_lora_weights(args.output_dir, adapter_name="cogvideox-lora")