Upload weights and config

README.md

@@ -0,0 +1,9 @@
+---
+tags:
+- model_hub_mixin
+- pytorch_model_hub_mixin
+---
+
+This model has been pushed to the Hub using the [PytorchModelHubMixin](https://huggingface.co/docs/huggingface_hub/package_reference/mixins#huggingface_hub.PyTorchModelHubMixin) integration:
+- Library: [More Information Needed]
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,35 @@
+{
+  "predictor": {
+    "_target_": "comit.components.predictor.RuledDiT",
+    "depth": 12,
+    "hidden_size": 768,
+    "in_channels": 4,
+    "max_input_size": 32,
+    "mlp_ratio": 4.0,
+    "msg_tokens_dim": 6,
+    "num_heads": 12,
+    "num_msg_tokens": 256,
+    "out_channels": 4,
+    "patch_size": 2,
+    "representation_layer": 4
+  },
+  "quantizer": {
+    "_target_": "comit.components.quantizer.FSQ",
+    "apply_corrupt_tokens_p": 0.0,
+    "corrupt_tokens_p": 0.0,
+    "drop_quant_p": 0.0,
+    "levels": [
+      8,
+      8,
+      8,
+      5,
+      5,
+      5
+    ],
+    "min_corrupt_tokens_p": null
+  },
+  "vae": {
+    "_target_": "comit.components.vae.SDAutoencoderKL",
+    "model_type": "ema"
+  }
+}

model.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76d3c728682308631957312bf2d5ad5d5a4c4256e3911205c9f3dec996df5ecc
+size 1030042076