From 56ed84acbdec3ecfbd2f6bc061a21a370d9f4e18 Mon Sep 17 00:00:00 2001 From: sennnnn Date: Mon, 11 Mar 2024 15:46:54 +0800 Subject: [PATCH] Add hidden_size argument for videogpt config. Former-commit-id: 1be9f22767fe9c80ad90cb17184751b97aff9012 --- opensora/models/ae/videobase/vqvae/configuration_vqvae.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/opensora/models/ae/videobase/vqvae/configuration_vqvae.py b/opensora/models/ae/videobase/vqvae/configuration_vqvae.py index 450735aeb..90ac29cfa 100644 --- a/opensora/models/ae/videobase/vqvae/configuration_vqvae.py +++ b/opensora/models/ae/videobase/vqvae/configuration_vqvae.py @@ -29,3 +29,5 @@ class VQVAEConfiguration(VideoBaseConfiguration): self.downsample = downsample self.no_pos_embd = no_pos_embd + + self.hidden_size = n_hiddens