Push model using huggingface_hub.

README.md

@@ -1,3 +1,14 @@
----
-license: mit
----
+---
+license: mit
+tags:
+- cad
+- diffusion
+- arxiv:2405.20324
+- pytorch_model_hub_mixin
+- model_hub_mixin
+pipeline_tag: text-to-image
+---
+
+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: https://github.com/nicolas-dufour/CAD
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,32 @@
+{
+  "bias_init": "zeros",
+  "compute_dropout": 0,
+  "compute_heads": 16,
+  "compute_stochastic_depth": 0,
+  "concat_cond_token_to_latents": false,
+  "data_dim": 256,
+  "data_mlp_multiplier": 4,
+  "data_positional_embedding_type": "learned",
+  "data_size": 64,
+  "label_dim": 2048,
+  "latent_mlp_multiplier": 4,
+  "latents_dim": 1024,
+  "modulate_conditioning": false,
+  "noise_embedding_type": "positional",
+  "num_blocks": 4,
+  "num_cond_tokens": 77,
+  "num_input_channels": 4,
+  "num_latents": 512,
+  "num_processing_layers": 4,
+  "num_text_registers": 16,
+  "path_size": 2,
+  "read_write_heads": 16,
+  "rw_dropout": 0.0,
+  "rw_stochastic_depth": 0,
+  "time_scaling": 1000.0,
+  "use_16_bits_layer_norm": false,
+  "use_biases": true,
+  "use_cond_rin_block": true,
+  "use_cond_token": true,
+  "weight_init": "xavier_uniform"
+}

model.safetensors

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