Add hidden_size argument for videogpt config.

Former-commit-id: 1be9f22767fe9c80ad90cb17184751b97aff9012
2024-03-11 15:46:54 +08:00
parent 9fbd96d66c
commit 56ed84acbd
1 changed files with 2 additions and 0 deletions
@@ -29,3 +29,5 @@ class VQVAEConfiguration(VideoBaseConfiguration):
            self.downsample = downsample

        self.no_pos_embd = no_pos_embd
+
+        self.hidden_size = n_hiddens