init

text_encoder/config.json

@@ -0,0 +1,25 @@
+{
+  "_name_or_path": "openMUSE/CLIP-ViT-L-14-DataComp.XL-s13B-b90K-penultimate",
+  "architectures": [
+    "CLIPTextModel"
+  ],
+  "attention_dropout": 0.0,
+  "bos_token_id": 0,
+  "dropout": 0.0,
+  "eos_token_id": 2,
+  "hidden_act": "gelu",
+  "hidden_size": 768,
+  "initializer_factor": 1.0,
+  "initializer_range": 0.02,
+  "intermediate_size": 3072,
+  "layer_norm_eps": 1e-05,
+  "max_position_embeddings": 77,
+  "model_type": "clip_text_model",
+  "num_attention_heads": 12,
+  "num_hidden_layers": 11,
+  "pad_token_id": 1,
+  "projection_dim": 512,
+  "torch_dtype": "float32",
+  "transformers_version": "4.26.1",
+  "vocab_size": 49408
+}

text_encoder/pytorch_model.bin

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

text_encoder/special_tokens_map.json

@@ -0,0 +1,24 @@
+{
+  "bos_token": {
+    "content": "<|startoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "eos_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": "<|endoftext|>",
+  "unk_token": {
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

text_encoder/tokenizer_config.json

@@ -0,0 +1,35 @@
+{
+  "add_prefix_space": false,
+  "bos_token": {
+    "__type": "AddedToken",
+    "content": "<|startoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "clean_up_tokenization_spaces": true,
+  "do_lower_case": true,
+  "eos_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  },
+  "errors": "replace",
+  "model_max_length": 77,
+  "name_or_path": "openMUSE/CLIP-ViT-L-14-DataComp.XL-s13B-b90K-penultimate",
+  "pad_token": "<|endoftext|>",
+  "special_tokens_map_file": "/home/william_huggingface_co/.cache/huggingface/hub/models--openMUSE--CLIP-ViT-L-14-DataComp.XL-s13B-b90K-penultimate/snapshots/9623448d1c3d9ce39bd28d5393c29cc08f85ba97/special_tokens_map.json",
+  "tokenizer_class": "CLIPTokenizer",
+  "unk_token": {
+    "__type": "AddedToken",
+    "content": "<|endoftext|>",
+    "lstrip": false,
+    "normalized": true,
+    "rstrip": false,
+    "single_word": false
+  }
+}

transformer/config.json

@@ -0,0 +1,34 @@
+{
+  "_class_name": "MaskGiTUViT",
+  "_version": "0.0.1",
+  "add_cross_attention": true,
+  "attention_dropout": 0.0,
+  "block_out_channels": [
+    512,
+    1024
+  ],
+  "codebook_size": 8192,
+  "encoder_hidden_size": 768,
+  "hidden_dropout": 0.0,
+  "hidden_size": 1024,
+  "in_channels": 384,
+  "initializer_range": 0.02,
+  "intermediate_size": 4096,
+  "layer_norm_eps": "1e-6",
+  "mask_token_id": 8255,
+  "max_position_embeddings": 256,
+  "norm_type": "rmsnorm",
+  "num_attention_heads": 16,
+  "num_classes": null,
+  "num_hidden_layers": 22,
+  "num_res_blocks": 3,
+  "num_vq_tokens": 1024,
+  "patch_size": 1,
+  "project_encoder_hidden_states": false,
+  "use_bias": false,
+  "use_codebook_size_for_output": true,
+  "use_encoder_layernorm": true,
+  "use_normformer": false,
+  "use_position_embeddings": false,
+  "vocab_size": 8256
+}

transformer/pytorch_model.bin

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

vae/config.json

@@ -0,0 +1,11 @@
+{
+  "_class_name": "PaellaVQModel",
+  "_name_or_path": "openMUSE/paellavq-f8-8192-laion",
+  "_version": "0.0.1",
+  "bottleneck_blocks": 12,
+  "c_hidden": 512,
+  "c_latent": 4,
+  "codebook_size": 8192,
+  "levels": 3,
+  "scale_factor": 0.3764
+}

vae/pytorch_model.bin

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