initial commit

.gitignore

@@ -0,0 +1,11 @@
+output/
+*__pycache__/
+samples*/
+runs/
+checkpoints/
+master_ip
+logs/
+*.DS_Store
+.idea
+*.pt
+tools/

__init__.py

@@ -0,0 +1,4 @@
+from .nodes import NODE_CLASS_MAPPINGS, NODE_DISPLAY_NAME_MAPPINGS
+
+
+__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]

configs/hy_vae_config.json

@@ -0,0 +1,34 @@
+{
+  "_class_name": "AutoencoderKLCausal3D",
+  "_diffusers_version": "0.4.2",
+  "act_fn": "silu",
+  "block_out_channels": [
+    128,
+    256,
+    512,
+    512
+  ],
+  "down_block_types": [
+    "DownEncoderBlockCausal3D",
+    "DownEncoderBlockCausal3D",
+    "DownEncoderBlockCausal3D",
+    "DownEncoderBlockCausal3D"
+  ],
+  "in_channels": 3,
+  "latent_channels": 16,
+  "layers_per_block": 2,
+  "norm_num_groups": 32,
+  "out_channels": 3,
+  "sample_size": 256,
+  "sample_tsize": 64,
+  "up_block_types": [
+    "UpDecoderBlockCausal3D",
+    "UpDecoderBlockCausal3D",
+    "UpDecoderBlockCausal3D",
+    "UpDecoderBlockCausal3D"
+  ],
+  "scaling_factor": 0.476986,
+  "time_compression_ratio": 4,
+  "mid_block_add_attention": true,
+  "mid_block_causal_attn": true
+}