From 9cb4aa16eba019f5235352db1a9012e673f18d58 Mon Sep 17 00:00:00 2001 From: Artiprocher Date: Wed, 20 Nov 2024 09:51:31 +0800 Subject: [PATCH] fix cogvideo height width checker --- diffsynth/pipelines/cog_video.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/diffsynth/pipelines/cog_video.py b/diffsynth/pipelines/cog_video.py index 4b7f336..f42d295 100644 --- a/diffsynth/pipelines/cog_video.py +++ b/diffsynth/pipelines/cog_video.py @@ -13,7 +13,7 @@ from einops import rearrange class CogVideoPipeline(BasePipeline): def __init__(self, device="cuda", torch_dtype=torch.float16): - super().__init__(device=device, torch_dtype=torch_dtype) + super().__init__(device=device, torch_dtype=torch_dtype, height_division_factor=16, width_division_factor=16) self.scheduler = EnhancedDDIMScheduler(rescale_zero_terminal_snr=True, prediction_type="v_prediction") self.prompter = CogPrompter() # models