Upload ToonCrafterDiffusionPipeline

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]

embedder/config.json

@@ -0,0 +1,10 @@
+{
+  "_class_name": "VisualTextCLIPEmbedder",
+  "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "workspace/tooncrafter/embedder",
+  "antialias": true,
+  "arch": "ViT-H-14",
+  "freeze": true,
+  "layer": "penultimate",
+  "version": null
+}

embedder/diffusion_pytorch_model.safetensors

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

image_resampler/config.json

@@ -0,0 +1,14 @@
+{
+  "_class_name": "ImageResampler",
+  "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "workspace/tooncrafter/image_resampler",
+  "depth": 4,
+  "dim": 1024,
+  "dim_head": 64,
+  "embedding_dim": 1280,
+  "ff_mult": 4,
+  "heads": 12,
+  "num_queries": 16,
+  "output_dim": 1024,
+  "video_length": 16
+}

image_resampler/diffusion_pytorch_model.safetensors

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

model_index.json

@@ -0,0 +1,25 @@
+{
+  "_class_name": "ToonCrafterDiffusionPipeline",
+  "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "workspace/tooncrafter",
+  "embedder": [
+    "modules.tooncrafter_condition",
+    "VisualTextCLIPEmbedder"
+  ],
+  "image_resampler": [
+    "modules.tooncrafter_condition",
+    "ImageResampler"
+  ],
+  "scheduler": [
+    "modules.scheduler",
+    "DDIMSchedulerVDM"
+  ],
+  "unet": [
+    "modules.tooncrafter_unet",
+    "UNet3DModel"
+  ],
+  "vae": [
+    "modules.tooncrafter_vae",
+    "VideoAutoencoderKL"
+  ]
+}

scheduler/scheduler_config.json

@@ -0,0 +1,22 @@
+{
+  "_class_name": "DDIMSchedulerVDM",
+  "_diffusers_version": "0.30.0.dev0",
+  "beta_end": 0.012,
+  "beta_schedule": "scaled_linear",
+  "beta_start": 0.00085,
+  "clip_sample": false,
+  "clip_sample_range": 1.0,
+  "dynamic_thresholding_ratio": 0.995,
+  "dynamic_tsnr_rescale": true,
+  "dynamic_tsnr_terminal_value": 0.7,
+  "dynamic_tsnr_turning_step": 400,
+  "num_train_timesteps": 1000,
+  "prediction_type": "v_prediction",
+  "rescale_betas_zero_snr": true,
+  "sample_max_value": 1.0,
+  "set_alpha_to_one": false,
+  "steps_offset": 0,
+  "thresholding": false,
+  "timestep_spacing": "trailing",
+  "trained_betas": null
+}

unet/config.json

@@ -0,0 +1,43 @@
+{
+  "_class_name": "UNet3DModel",
+  "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "workspace/tooncrafter/unet",
+  "addition_attention": true,
+  "attention_resolutions": [
+    4,
+    2,
+    1
+  ],
+  "channel_mult": [
+    1,
+    2,
+    4,
+    4
+  ],
+  "context_dim": 1024,
+  "conv_resample": true,
+  "default_fs": 24,
+  "dims": 2,
+  "dropout": 0.1,
+  "fs_condition": true,
+  "image_cross_attention": true,
+  "image_cross_attention_scale_learnable": false,
+  "in_channels": 8,
+  "model_channels": 320,
+  "num_head_channels": 64,
+  "num_heads": -1,
+  "num_res_blocks": 2,
+  "out_channels": 4,
+  "resblock_updown": false,
+  "temporal_attention": true,
+  "temporal_conv": true,
+  "temporal_length": 16,
+  "temporal_selfatt_only": true,
+  "tempspatial_aware": false,
+  "transformer_depth": 1,
+  "use_causal_attention": false,
+  "use_checkpoint": true,
+  "use_linear": true,
+  "use_relative_position": false,
+  "use_scale_shift_norm": false
+}

unet/diffusion_pytorch_model.safetensors

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

vae/config.json

@@ -0,0 +1,21 @@
+{
+  "_class_name": "VideoAutoencoderKL",
+  "_diffusers_version": "0.30.0.dev0",
+  "_name_or_path": "workspace/tooncrafter/vae",
+  "attn_resolutions": [],
+  "ch": 128,
+  "ch_mult": [
+    1,
+    2,
+    4,
+    4
+  ],
+  "double_z": true,
+  "dropout": 0.0,
+  "in_channels": 3,
+  "num_res_blocks": 2,
+  "out_ch": 3,
+  "resolution": 256,
+  "scaling_factor": 0.18215,
+  "z_channels": 4
+}

vae/diffusion_pytorch_model.safetensors

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