Update model

README.md

@@ -1,3 +1,14 @@
----
-license: mit
----
+---
+license: mit
+pipeline_tag: text-to-image
+tags:
+- ImageNet
+- arxiv:2502.21318
+- diffusion
+- 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: https://github.com/lucasdegeorge/T2I-ImageNet/
+- Docs: [More Information Needed]

config.json

@@ -0,0 +1,31 @@
+{
+  "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": 32,
+  "label_dim": 2048,
+  "latent_mlp_multiplier": 4,
+  "latents_dim": 1024,
+  "noise_embedding_type": "positional",
+  "num_blocks": 4,
+  "num_cond_tokens": 77,
+  "num_input_channels": 4,
+  "num_latents": 256,
+  "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:d67d35eb8271f654d99ce226c5219b6bfce2ebd1c153717c9abd28a1701451d6
+size 1326247177