initial commit

README.md

@@ -0,0 +1,198 @@
+---
+library_name: diffusers
+---
+
+# Model Card for Model ID
+
+<!-- Provide a quick summary of what the model is/does. -->
+
+
+
+## Model Details
+
+### Model Description
+
+<!-- Provide a longer summary of what this model is. -->
+
+This is the model card of a 🧨 diffusers model that has been pushed on the Hub. This model card has been automatically generated.
+
+- **Developed by:** [More Information Needed]
+- **Funded by [optional]:** [More Information Needed]
+- **Shared by [optional]:** [More Information Needed]
+- **Model type:** [More Information Needed]
+- **Language(s) (NLP):** [More Information Needed]
+- **License:** [More Information Needed]
+- **Finetuned from model [optional]:** [More Information Needed]
+
+### Model Sources [optional]
+
+<!-- Provide the basic links for the model. -->
+
+- **Repository:** [More Information Needed]
+- **Paper [optional]:** [More Information Needed]
+- **Demo [optional]:** [More Information Needed]
+
+## Uses
+
+<!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
+
+### Direct Use
+
+<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+
+[More Information Needed]
+
+### Downstream Use [optional]
+
+<!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
+
+[More Information Needed]
+
+### Out-of-Scope Use
+
+<!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
+
+[More Information Needed]
+
+## Bias, Risks, and Limitations
+
+<!-- This section is meant to convey both technical and sociotechnical limitations. -->
+
+[More Information Needed]
+
+### Recommendations
+
+<!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
+
+Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
+
+## How to Get Started with the Model
+
+Use the code below to get started with the model.
+
+[More Information Needed]
+
+## Training Details
+
+### Training Data
+
+<!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
+
+[More Information Needed]
+
+### Training Procedure
+
+<!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
+
+#### Preprocessing [optional]
+
+[More Information Needed]
+
+
+#### Training Hyperparameters
+
+- **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
+
+#### Speeds, Sizes, Times [optional]
+
+<!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
+
+[More Information Needed]
+
+## Evaluation
+
+<!-- This section describes the evaluation protocols and provides the results. -->
+
+### Testing Data, Factors & Metrics
+
+#### Testing Data
+
+<!-- This should link to a Dataset Card if possible. -->
+
+[More Information Needed]
+
+#### Factors
+
+<!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
+
+[More Information Needed]
+
+#### Metrics
+
+<!-- These are the evaluation metrics being used, ideally with a description of why. -->
+
+[More Information Needed]
+
+### Results
+
+[More Information Needed]
+
+#### Summary
+
+
+
+## Model Examination [optional]
+
+<!-- Relevant interpretability work for the model goes here -->
+
+[More Information Needed]
+
+## Environmental Impact
+
+<!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
+
+Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
+
+- **Hardware Type:** [More Information Needed]
+- **Hours used:** [More Information Needed]
+- **Cloud Provider:** [More Information Needed]
+- **Compute Region:** [More Information Needed]
+- **Carbon Emitted:** [More Information Needed]
+
+## Technical Specifications [optional]
+
+### Model Architecture and Objective
+
+[More Information Needed]
+
+### Compute Infrastructure
+
+[More Information Needed]
+
+#### Hardware
+
+[More Information Needed]
+
+#### Software
+
+[More Information Needed]
+
+## Citation [optional]
+
+<!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
+
+**BibTeX:**
+
+[More Information Needed]
+
+**APA:**
+
+[More Information Needed]
+
+## Glossary [optional]
+
+<!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
+
+[More Information Needed]
+
+## More Information [optional]
+
+[More Information Needed]
+
+## Model Card Authors [optional]
+
+[More Information Needed]
+
+## Model Card Contact
+
+[More Information Needed]

model_index.json

@@ -0,0 +1,29 @@
+{
+  "_class_name": "CosmosTextToWorldPipeline",
+  "_diffusers_version": "0.34.0.dev0",
+  "_name_or_path": "tiny-cosmos",
+  "safety_checker": [
+    null,
+    null
+  ],
+  "scheduler": [
+    "diffusers",
+    "EDMEulerScheduler"
+  ],
+  "text_encoder": [
+    "transformers",
+    "T5EncoderModel"
+  ],
+  "tokenizer": [
+    "transformers",
+    "T5TokenizerFast"
+  ],
+  "transformer": [
+    "diffusers",
+    "CosmosTransformer3DModel"
+  ],
+  "vae": [
+    "diffusers",
+    "AutoencoderKLCosmos"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,12 @@
+{
+  "_class_name": "EDMEulerScheduler",
+  "_diffusers_version": "0.34.0.dev0",
+  "final_sigmas_type": "sigma_min",
+  "num_train_timesteps": 1000,
+  "prediction_type": "epsilon",
+  "rho": 7.0,
+  "sigma_data": 0.5,
+  "sigma_max": 80,
+  "sigma_min": 0.002,
+  "sigma_schedule": "karras"
+}

text_encoder/config.json

@@ -0,0 +1,60 @@
+{
+  "architectures": [
+    "T5EncoderModel"
+  ],
+  "classifier_dropout": 0.0,
+  "d_ff": 65536,
+  "d_kv": 64,
+  "d_model": 64,
+  "decoder_start_token_id": 0,
+  "dense_act_fn": "relu",
+  "dropout_rate": 0.1,
+  "eos_token_id": 1,
+  "feed_forward_proj": "relu",
+  "initializer_factor": 1.0,
+  "is_encoder_decoder": true,
+  "is_gated_act": false,
+  "layer_norm_epsilon": 1e-06,
+  "model_type": "t5",
+  "n_positions": 512,
+  "num_decoder_layers": 1,
+  "num_heads": 128,
+  "num_layers": 1,
+  "output_past": true,
+  "pad_token_id": 0,
+  "relative_attention_max_distance": 128,
+  "relative_attention_num_buckets": 32,
+  "task_specific_params": {
+    "summarization": {
+      "early_stopping": true,
+      "length_penalty": 2.0,
+      "max_length": 200,
+      "min_length": 30,
+      "no_repeat_ngram_size": 3,
+      "num_beams": 4,
+      "prefix": "summarize: "
+    },
+    "translation_en_to_de": {
+      "early_stopping": true,
+      "max_length": 300,
+      "num_beams": 4,
+      "prefix": "translate English to German: "
+    },
+    "translation_en_to_fr": {
+      "early_stopping": true,
+      "max_length": 300,
+      "num_beams": 4,
+      "prefix": "translate English to French: "
+    },
+    "translation_en_to_ro": {
+      "early_stopping": true,
+      "max_length": 300,
+      "num_beams": 4,
+      "prefix": "translate English to Romanian: "
+    }
+  },
+  "torch_dtype": "float32",
+  "transformers_version": "4.51.3",
+  "use_cache": true,
+  "vocab_size": 32128
+}

text_encoder/model.safetensors

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

tokenizer/special_tokens_map.json

@@ -0,0 +1,125 @@
+{
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "eos_token": {
+    "content": "</s>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "pad_token": {
+    "content": "<pad>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  },
+  "unk_token": {
+    "content": "<unk>",
+    "lstrip": false,
+    "normalized": false,
+    "rstrip": false,
+    "single_word": false
+  }
+}

tokenizer/spiece.model

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

tokenizer/tokenizer_config.json

@@ -0,0 +1,939 @@
+{
+  "add_prefix_space": null,
+  "added_tokens_decoder": {
+    "0": {
+      "content": "<pad>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "1": {
+      "content": "</s>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "2": {
+      "content": "<unk>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32000": {
+      "content": "<extra_id_99>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32001": {
+      "content": "<extra_id_98>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32002": {
+      "content": "<extra_id_97>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32003": {
+      "content": "<extra_id_96>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32004": {
+      "content": "<extra_id_95>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32005": {
+      "content": "<extra_id_94>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32006": {
+      "content": "<extra_id_93>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32007": {
+      "content": "<extra_id_92>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32008": {
+      "content": "<extra_id_91>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32009": {
+      "content": "<extra_id_90>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32010": {
+      "content": "<extra_id_89>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32011": {
+      "content": "<extra_id_88>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32012": {
+      "content": "<extra_id_87>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32013": {
+      "content": "<extra_id_86>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32014": {
+      "content": "<extra_id_85>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32015": {
+      "content": "<extra_id_84>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32016": {
+      "content": "<extra_id_83>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32017": {
+      "content": "<extra_id_82>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32018": {
+      "content": "<extra_id_81>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32019": {
+      "content": "<extra_id_80>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32020": {
+      "content": "<extra_id_79>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32021": {
+      "content": "<extra_id_78>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32022": {
+      "content": "<extra_id_77>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32023": {
+      "content": "<extra_id_76>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32024": {
+      "content": "<extra_id_75>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32025": {
+      "content": "<extra_id_74>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32026": {
+      "content": "<extra_id_73>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32027": {
+      "content": "<extra_id_72>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32028": {
+      "content": "<extra_id_71>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32029": {
+      "content": "<extra_id_70>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32030": {
+      "content": "<extra_id_69>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32031": {
+      "content": "<extra_id_68>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32032": {
+      "content": "<extra_id_67>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32033": {
+      "content": "<extra_id_66>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32034": {
+      "content": "<extra_id_65>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32035": {
+      "content": "<extra_id_64>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32036": {
+      "content": "<extra_id_63>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32037": {
+      "content": "<extra_id_62>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32038": {
+      "content": "<extra_id_61>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32039": {
+      "content": "<extra_id_60>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32040": {
+      "content": "<extra_id_59>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32041": {
+      "content": "<extra_id_58>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32042": {
+      "content": "<extra_id_57>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32043": {
+      "content": "<extra_id_56>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32044": {
+      "content": "<extra_id_55>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32045": {
+      "content": "<extra_id_54>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32046": {
+      "content": "<extra_id_53>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32047": {
+      "content": "<extra_id_52>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32048": {
+      "content": "<extra_id_51>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32049": {
+      "content": "<extra_id_50>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32050": {
+      "content": "<extra_id_49>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32051": {
+      "content": "<extra_id_48>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32052": {
+      "content": "<extra_id_47>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32053": {
+      "content": "<extra_id_46>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32054": {
+      "content": "<extra_id_45>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32055": {
+      "content": "<extra_id_44>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32056": {
+      "content": "<extra_id_43>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32057": {
+      "content": "<extra_id_42>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32058": {
+      "content": "<extra_id_41>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32059": {
+      "content": "<extra_id_40>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32060": {
+      "content": "<extra_id_39>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32061": {
+      "content": "<extra_id_38>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32062": {
+      "content": "<extra_id_37>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32063": {
+      "content": "<extra_id_36>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32064": {
+      "content": "<extra_id_35>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32065": {
+      "content": "<extra_id_34>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32066": {
+      "content": "<extra_id_33>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32067": {
+      "content": "<extra_id_32>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32068": {
+      "content": "<extra_id_31>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32069": {
+      "content": "<extra_id_30>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32070": {
+      "content": "<extra_id_29>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32071": {
+      "content": "<extra_id_28>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32072": {
+      "content": "<extra_id_27>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32073": {
+      "content": "<extra_id_26>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32074": {
+      "content": "<extra_id_25>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32075": {
+      "content": "<extra_id_24>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32076": {
+      "content": "<extra_id_23>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32077": {
+      "content": "<extra_id_22>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32078": {
+      "content": "<extra_id_21>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32079": {
+      "content": "<extra_id_20>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32080": {
+      "content": "<extra_id_19>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32081": {
+      "content": "<extra_id_18>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32082": {
+      "content": "<extra_id_17>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32083": {
+      "content": "<extra_id_16>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32084": {
+      "content": "<extra_id_15>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32085": {
+      "content": "<extra_id_14>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32086": {
+      "content": "<extra_id_13>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32087": {
+      "content": "<extra_id_12>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32088": {
+      "content": "<extra_id_11>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32089": {
+      "content": "<extra_id_10>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32090": {
+      "content": "<extra_id_9>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32091": {
+      "content": "<extra_id_8>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32092": {
+      "content": "<extra_id_7>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32093": {
+      "content": "<extra_id_6>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32094": {
+      "content": "<extra_id_5>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32095": {
+      "content": "<extra_id_4>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32096": {
+      "content": "<extra_id_3>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32097": {
+      "content": "<extra_id_2>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32098": {
+      "content": "<extra_id_1>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    },
+    "32099": {
+      "content": "<extra_id_0>",
+      "lstrip": false,
+      "normalized": false,
+      "rstrip": false,
+      "single_word": false,
+      "special": true
+    }
+  },
+  "additional_special_tokens": [
+    "<extra_id_0>",
+    "<extra_id_1>",
+    "<extra_id_2>",
+    "<extra_id_3>",
+    "<extra_id_4>",
+    "<extra_id_5>",
+    "<extra_id_6>",
+    "<extra_id_7>",
+    "<extra_id_8>",
+    "<extra_id_9>",
+    "<extra_id_10>",
+    "<extra_id_11>",
+    "<extra_id_12>",
+    "<extra_id_13>",
+    "<extra_id_14>",
+    "<extra_id_15>",
+    "<extra_id_16>",
+    "<extra_id_17>",
+    "<extra_id_18>",
+    "<extra_id_19>",
+    "<extra_id_20>",
+    "<extra_id_21>",
+    "<extra_id_22>",
+    "<extra_id_23>",
+    "<extra_id_24>",
+    "<extra_id_25>",
+    "<extra_id_26>",
+    "<extra_id_27>",
+    "<extra_id_28>",
+    "<extra_id_29>",
+    "<extra_id_30>",
+    "<extra_id_31>",
+    "<extra_id_32>",
+    "<extra_id_33>",
+    "<extra_id_34>",
+    "<extra_id_35>",
+    "<extra_id_36>",
+    "<extra_id_37>",
+    "<extra_id_38>",
+    "<extra_id_39>",
+    "<extra_id_40>",
+    "<extra_id_41>",
+    "<extra_id_42>",
+    "<extra_id_43>",
+    "<extra_id_44>",
+    "<extra_id_45>",
+    "<extra_id_46>",
+    "<extra_id_47>",
+    "<extra_id_48>",
+    "<extra_id_49>",
+    "<extra_id_50>",
+    "<extra_id_51>",
+    "<extra_id_52>",
+    "<extra_id_53>",
+    "<extra_id_54>",
+    "<extra_id_55>",
+    "<extra_id_56>",
+    "<extra_id_57>",
+    "<extra_id_58>",
+    "<extra_id_59>",
+    "<extra_id_60>",
+    "<extra_id_61>",
+    "<extra_id_62>",
+    "<extra_id_63>",
+    "<extra_id_64>",
+    "<extra_id_65>",
+    "<extra_id_66>",
+    "<extra_id_67>",
+    "<extra_id_68>",
+    "<extra_id_69>",
+    "<extra_id_70>",
+    "<extra_id_71>",
+    "<extra_id_72>",
+    "<extra_id_73>",
+    "<extra_id_74>",
+    "<extra_id_75>",
+    "<extra_id_76>",
+    "<extra_id_77>",
+    "<extra_id_78>",
+    "<extra_id_79>",
+    "<extra_id_80>",
+    "<extra_id_81>",
+    "<extra_id_82>",
+    "<extra_id_83>",
+    "<extra_id_84>",
+    "<extra_id_85>",
+    "<extra_id_86>",
+    "<extra_id_87>",
+    "<extra_id_88>",
+    "<extra_id_89>",
+    "<extra_id_90>",
+    "<extra_id_91>",
+    "<extra_id_92>",
+    "<extra_id_93>",
+    "<extra_id_94>",
+    "<extra_id_95>",
+    "<extra_id_96>",
+    "<extra_id_97>",
+    "<extra_id_98>",
+    "<extra_id_99>"
+  ],
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "</s>",
+  "extra_ids": 100,
+  "extra_special_tokens": {},
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "tokenizer_class": "T5Tokenizer",
+  "unk_token": "<unk>"
+}

transformer/config.json

@@ -0,0 +1,30 @@
+{
+  "_class_name": "CosmosTransformer3DModel",
+  "_diffusers_version": "0.34.0.dev0",
+  "_name_or_path": "tiny-cosmos/transformer",
+  "adaln_lora_dim": 256,
+  "attention_head_dim": 128,
+  "concat_padding_mask": true,
+  "extra_pos_embed_type": "learnable",
+  "in_channels": 16,
+  "max_size": [
+    128,
+    240,
+    240
+  ],
+  "mlp_ratio": 4.0,
+  "num_attention_heads": 2,
+  "num_layers": 2,
+  "out_channels": 16,
+  "patch_size": [
+    1,
+    2,
+    2
+  ],
+  "rope_scale": [
+    2.0,
+    1.0,
+    1.0
+  ],
+  "text_embed_dim": 64
+}

transformer/diffusion_pytorch_model.safetensors

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

vae/config.json

@@ -0,0 +1,546 @@
+{
+  "_class_name": "AutoencoderKLCosmos",
+  "_diffusers_version": "0.34.0.dev0",
+  "_name_or_path": "tiny-cosmos/vae",
+  "attention_resolutions": [
+    32
+  ],
+  "decode_block_out_channels": [
+    8,
+    16,
+    16,
+    16
+  ],
+  "encoder_block_out_channels": [
+    4,
+    8,
+    16,
+    16
+  ],
+  "in_channels": 3,
+  "latent_channels": 16,
+  "latents_mean": [
+    0.11362757533788681,
+    -0.017171695828437805,
+    0.03071162849664688,
+    0.020468616858124733,
+    0.019314559176564217,
+    0.02138567343354225,
+    0.019993416965007782,
+    0.021891867741942406,
+    0.020119348540902138,
+    0.01872694492340088,
+    0.021686134859919548,
+    0.022071482613682747,
+    0.01986941136419773,
+    0.017704127356410027,
+    0.020676426589488983,
+    0.0202824454754591,
+    0.19125476479530334,
+    0.04556972160935402,
+    0.05955580249428749,
+    0.05315534397959709,
+    0.05496629327535629,
+    0.05356264486908913,
+    0.04856595769524574,
+    0.05327453464269638,
+    0.05410471558570862,
+    0.05597148835659027,
+    0.05524865910410881,
+    0.051818735897541046,
+    0.05071663111448288,
+    0.052045371383428574,
+    0.05641080439090729,
+    0.05518041551113129,
+    0.01306714117527008,
+    0.03341160714626312,
+    0.03847246244549751,
+    0.028101854026317596,
+    0.027901655063033104,
+    0.02920026145875454,
+    0.02823597379028797,
+    0.026310328394174576,
+    0.02785310335457325,
+    0.028805073350667953,
+    0.02977769263088703,
+    0.031454406678676605,
+    0.02888389490544796,
+    0.03280773386359215,
+    0.03484927490353584,
+    0.03049197606742382,
+    -0.0019772653467953205,
+    0.07534956932067871,
+    0.049638792872428894,
+    0.05530892685055733,
+    0.054108284413814545,
+    0.052525412291288376,
+    0.05029899254441261,
+    0.05321024730801582,
+    0.05149244889616966,
+    0.05119210109114647,
+    0.04643494635820389,
+    0.04604526609182358,
+    0.046316176652908325,
+    0.04404101148247719,
+    0.04403536394238472,
+    0.04499494656920433,
+    -0.029941828921437263,
+    -0.047870032489299774,
+    -0.010645583271980286,
+    -0.01779823563992977,
+    -0.014905018731951714,
+    -0.02157517336308956,
+    -0.02047779969871044,
+    -0.021808160468935966,
+    -0.0194537490606308,
+    -0.020628631114959717,
+    -0.021922089159488678,
+    -0.02520638518035412,
+    -0.022466560825705528,
+    -0.024275334551930428,
+    -0.02683362551033497,
+    -0.02762005850672722,
+    0.08019473403692245,
+    -0.13005368411540985,
+    -0.07568635791540146,
+    -0.06082374230027199,
+    -0.060361746698617935,
+    -0.058753639459609985,
+    -0.0592188723385334,
+    -0.058697883039712906,
+    -0.05273940786719322,
+    -0.05256499722599983,
+    -0.05346428230404854,
+    -0.05456540733575821,
+    -0.05365699902176857,
+    -0.05656896531581879,
+    -0.057285893708467484,
+    -0.05321847274899483,
+    0.1671840250492096,
+    -0.003901461837813258,
+    0.03794059902429581,
+    0.03565610200166702,
+    0.03554131090641022,
+    0.03924074023962021,
+    0.03873615339398384,
+    0.04187328740954399,
+    0.04226924106478691,
+    0.04378717020153999,
+    0.04684273898601532,
+    0.05117613822221756,
+    0.04547792300581932,
+    0.05251586064696312,
+    0.05048338696360588,
+    0.04950783774256706,
+    0.09564418345689774,
+    0.05471280217170715,
+    0.08183968812227249,
+    0.07978633046150208,
+    0.08076023310422897,
+    0.08108604699373245,
+    0.08011817932128906,
+    0.0796557292342186,
+    0.08187773078680038,
+    0.0835026279091835,
+    0.08101469278335571,
+    0.07869409769773483,
+    0.07744419574737549,
+    0.07724520564079285,
+    0.07830417901277542,
+    0.0759979635477066,
+    -0.049875665456056595,
+    0.05923908203840256,
+    -0.010587457567453384,
+    -0.011776033788919449,
+    -0.01116162072867155,
+    -0.013641487807035446,
+    -0.015460136346518993,
+    -0.01172130461782217,
+    -0.017800433561205864,
+    -0.016483137384057045,
+    -0.02100246772170067,
+    -0.0210441704839468,
+    -0.02482122741639614,
+    -0.02611689455807209,
+    -0.025611430406570435,
+    -0.025973355397582054,
+    -0.053646672517061234,
+    0.08211684226989746,
+    0.046869371086359024,
+    0.04605640843510628,
+    0.043041862547397614,
+    0.039735499769449234,
+    0.036867666989564896,
+    0.04087112471461296,
+    0.03704741224646568,
+    0.03706400841474533,
+    0.031200725585222244,
+    0.03349090740084648,
+    0.03319963067770004,
+    0.032057810574769974,
+    0.03195127099752426,
+    0.03180480748414993,
+    0.16427966952323914,
+    -0.1104845330119133,
+    -0.04595276340842247,
+    -0.04982893168926239,
+    -0.05213464796543121,
+    -0.04809378460049629,
+    -0.050803184509277344,
+    -0.04992862790822983,
+    -0.044933367520570755,
+    -0.046761900186538696,
+    -0.048847030848264694,
+    -0.04627892002463341,
+    -0.04913311451673508,
+    -0.04955708608031273,
+    -0.04533981531858444,
+    -0.04570218175649643,
+    -0.10612928122282028,
+    -0.05121197551488876,
+    -0.06761008501052856,
+    -0.0725180059671402,
+    -0.07265284657478333,
+    -0.074178546667099,
+    -0.07202412188053131,
+    -0.07499026507139206,
+    -0.07625480741262436,
+    -0.07535747438669205,
+    -0.07638787478208542,
+    -0.0792030468583107,
+    -0.0759606882929802,
+    -0.07959417998790741,
+    -0.08265036344528198,
+    -0.07955470681190491,
+    -0.1688891500234604,
+    0.07532420009374619,
+    0.04062594473361969,
+    0.03375093266367912,
+    0.033374518156051636,
+    0.036993756890296936,
+    0.0365113765001297,
+    0.036110229790210724,
+    0.03555621951818466,
+    0.03378554433584213,
+    0.030049804598093033,
+    0.033955592662096024,
+    0.029418472200632095,
+    0.031564321368932724,
+    0.034311726689338684,
+    0.030168525874614716,
+    -0.03415358439087868,
+    -0.016995728015899658,
+    -0.040292952209711075,
+    -0.049121566116809845,
+    -0.04988580197095871,
+    -0.04917917773127556,
+    -0.049180563539266586,
+    -0.052518896758556366,
+    -0.05325506255030632,
+    -0.05341973155736923,
+    -0.04983328655362129,
+    -0.048831455409526825,
+    -0.04985548183321953,
+    -0.04736718162894249,
+    -0.04620270058512688,
+    -0.0483609102666378,
+    0.020556746050715446,
+    0.03419799357652664,
+    -0.029076693579554558,
+    -0.04350508749485016,
+    -0.04156143590807915,
+    -0.04234420508146286,
+    -0.04446109011769295,
+    -0.04461774230003357,
+    -0.04882838577032089,
+    -0.04822345823049545,
+    -0.04502492770552635,
+    -0.05062440410256386,
+    -0.05146912857890129,
+    -0.046552665531635284,
+    -0.048629943281412125,
+    -0.048416152596473694,
+    0.20312774181365967,
+    -0.07208502292633057,
+    -0.03635615482926369,
+    -0.03556087613105774,
+    -0.042461737990379333,
+    -0.04195837676525116,
+    -0.04293777793645859,
+    -0.04071275517344475,
+    -0.042405690997838974,
+    -0.04125212877988815,
+    -0.04395143687725067,
+    -0.03959095850586891,
+    -0.04044992849230766,
+    -0.040158748626708984,
+    -0.04088107496500015,
+    -0.03885176032781601
+  ],
+  "latents_std": [
+    0.5670027136802673,
+    0.6548898220062256,
+    0.6558942794799805,
+    0.6652436852455139,
+    0.6661978363990784,
+    0.666638195514679,
+    0.6720837950706482,
+    0.6695597767829895,
+    0.6692887544631958,
+    0.6710878610610962,
+    0.6709252595901489,
+    0.6739746332168579,
+    0.6789488196372986,
+    0.6766831278800964,
+    0.6776956915855408,
+    0.6747955679893494,
+    0.8524512052536011,
+    0.868837296962738,
+    0.8734808564186096,
+    0.8845933675765991,
+    0.8913588523864746,
+    0.891050398349762,
+    0.8971490859985352,
+    0.8994747400283813,
+    0.9020176529884338,
+    0.9041182398796082,
+    0.906926155090332,
+    0.9084777235984802,
+    0.9064871072769165,
+    0.9100698232650757,
+    0.9103343486785889,
+    0.9054154753684998,
+    0.8496035933494568,
+    0.8586335182189941,
+    0.8689531683921814,
+    0.884606122970581,
+    0.8924500346183777,
+    0.8945170640945435,
+    0.8993100523948669,
+    0.9064735770225525,
+    0.9033823609352112,
+    0.9051007628440857,
+    0.9100831151008606,
+    0.9096121788024902,
+    0.9123716950416565,
+    0.9131317138671875,
+    0.914355456829071,
+    0.9156510233879089,
+    0.9187710285186768,
+    0.851551353931427,
+    0.857804000377655,
+    0.8699803352355957,
+    0.8736526370048523,
+    0.8816176652908325,
+    0.881510317325592,
+    0.887589156627655,
+    0.8901551365852356,
+    0.8924557566642761,
+    0.8927622437477112,
+    0.894504964351654,
+    0.9005420207977295,
+    0.8999413251876831,
+    0.9013610482215881,
+    0.9011489152908325,
+    0.7775575518608093,
+    0.8145685195922852,
+    0.8191184401512146,
+    0.8313707113265991,
+    0.8382047414779663,
+    0.8389037251472473,
+    0.8440110087394714,
+    0.844251811504364,
+    0.8473935723304749,
+    0.8479875326156616,
+    0.8524958491325378,
+    0.8511499762535095,
+    0.8516093492507935,
+    0.8562635779380798,
+    0.856778621673584,
+    0.8564102649688721,
+    0.6990351676940918,
+    0.7169788479804993,
+    0.7169691324234009,
+    0.7258316874504089,
+    0.729317307472229,
+    0.7325412631034851,
+    0.7358697652816772,
+    0.7373496890068054,
+    0.7366458177566528,
+    0.7408490777015686,
+    0.7439932227134705,
+    0.7447181940078735,
+    0.7449318766593933,
+    0.7482457756996155,
+    0.7502487301826477,
+    0.7527480125427246,
+    0.8187142014503479,
+    0.8225188255310059,
+    0.8261602520942688,
+    0.8316448330879211,
+    0.8407237529754639,
+    0.8396466970443726,
+    0.841433048248291,
+    0.848807692527771,
+    0.8503468036651611,
+    0.8519694805145264,
+    0.8521105051040649,
+    0.8538666367530823,
+    0.8541001677513123,
+    0.8543934226036072,
+    0.8584784865379333,
+    0.8538527488708496,
+    0.6758398413658142,
+    0.6825984716415405,
+    0.6919885277748108,
+    0.6992884278297424,
+    0.7019432783126831,
+    0.7046700119972229,
+    0.7075554728507996,
+    0.7091785669326782,
+    0.7100769877433777,
+    0.7096350193023682,
+    0.7106407880783081,
+    0.7102733254432678,
+    0.7129116654396057,
+    0.7153753638267517,
+    0.7190250754356384,
+    0.716041624546051,
+    0.7245098948478699,
+    0.719799280166626,
+    0.7205737829208374,
+    0.7303546071052551,
+    0.7332962155342102,
+    0.7366002798080444,
+    0.7389146089553833,
+    0.7427999377250671,
+    0.7410569190979004,
+    0.7400243282318115,
+    0.7425758838653564,
+    0.7441611886024475,
+    0.7454389929771423,
+    0.7469444274902344,
+    0.7474706172943115,
+    0.7458640336990356,
+    0.9017698764801025,
+    0.9099067449569702,
+    0.9110680222511292,
+    0.9216378331184387,
+    0.9239023327827454,
+    0.9305619597434998,
+    0.9348220229148865,
+    0.9364241361618042,
+    0.9385837912559509,
+    0.9406497478485107,
+    0.9407893419265747,
+    0.9432571530342102,
+    0.9495530128479004,
+    0.9481470584869385,
+    0.95144122838974,
+    0.9492307305335999,
+    0.49853548407554626,
+    0.6496810913085938,
+    0.6427654027938843,
+    0.6496639251708984,
+    0.6487663984298706,
+    0.6520355939865112,
+    0.6584241986274719,
+    0.6535161137580872,
+    0.654643714427948,
+    0.6574859023094177,
+    0.6562633514404297,
+    0.6612374782562256,
+    0.6612117886543274,
+    0.6607794165611267,
+    0.6604015231132507,
+    0.6647490859031677,
+    0.619865894317627,
+    0.6913813352584839,
+    0.6884557008743286,
+    0.6955842971801758,
+    0.6976540088653564,
+    0.7001534700393677,
+    0.705295979976654,
+    0.7046875357627869,
+    0.7039952278137207,
+    0.7047998905181885,
+    0.7088757157325745,
+    0.7112686634063721,
+    0.7097226977348328,
+    0.7124993205070496,
+    0.7123194932937622,
+    0.7117560505867004,
+    0.35586974024772644,
+    0.6872385740280151,
+    0.6897321939468384,
+    0.6995847821235657,
+    0.6943452954292297,
+    0.6995818018913269,
+    0.7098021507263184,
+    0.6989945769309998,
+    0.7027168869972229,
+    0.700950562953949,
+    0.6991285085678101,
+    0.7052269577980042,
+    0.7039217352867126,
+    0.7091691493988037,
+    0.7058573365211487,
+    0.7037354111671448,
+    0.9810133576393127,
+    0.890247642993927,
+    0.896072506904602,
+    0.906781792640686,
+    0.9130866527557373,
+    0.9181234836578369,
+    0.9198082685470581,
+    0.924806535243988,
+    0.9263566732406616,
+    0.92887943983078,
+    0.9333807229995728,
+    0.9346809387207031,
+    0.9361943602561951,
+    0.939060628414154,
+    0.9419177174568176,
+    0.9447172284126282,
+    0.832027792930603,
+    0.8410623073577881,
+    0.8446363210678101,
+    0.8582950830459595,
+    0.863196611404419,
+    0.8675134181976318,
+    0.8691433668136597,
+    0.8708592057228088,
+    0.8728635907173157,
+    0.8753739595413208,
+    0.8793113827705383,
+    0.8805447816848755,
+    0.8811838030815125,
+    0.8887255787849426,
+    0.889424741268158,
+    0.8893482685089111,
+    0.440253347158432,
+    0.6306161284446716,
+    0.6311061382293701,
+    0.6360195875167847,
+    0.6395812034606934,
+    0.6410434246063232,
+    0.650199294090271,
+    0.650279700756073,
+    0.643559455871582,
+    0.6465720534324646,
+    0.648470938205719,
+    0.6472811698913574,
+    0.6497294306755066,
+    0.6516297459602356,
+    0.6532804369926453,
+    0.6491477489471436
+  ],
+  "num_layers": 2,
+  "out_channels": 3,
+  "patch_size": 4,
+  "patch_type": "haar",
+  "resolution": 1024,
+  "scaling_factor": 1.0,
+  "spatial_compression_ratio": 8,
+  "temporal_compression_ratio": 8
+}

vae/diffusion_pytorch_model.safetensors

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