Lawrence-cj commited on
Commit
dea586e
·
verified ·
1 Parent(s): 6e876bc

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +104 -0
README.md ADDED
@@ -0,0 +1,104 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: sana
3
+ tags:
4
+ - text-to-image
5
+ - Sana
6
+ - 1024px_based_image_size
7
+ - Multi-language
8
+ - ControlNet
9
+ language:
10
+ - en
11
+ - zh
12
+ base_model:
13
+ - Efficient-Large-Model/Sana_600M_1024px_ControlNet_HED
14
+ pipeline_tag: text-to-image
15
+ ---
16
+
17
+ <p align="center" style="border-radius: 10px">
18
+ <img src="https://raw.githubusercontent.com/NVlabs/Sana/refs/heads/main/asset/logo.png" width="35%" alt="logo"/>
19
+ </p>
20
+
21
+ <div style="display:flex;justify-content: center">
22
+ <a href="https://huggingface.co/collections/Efficient-Large-Model/sana-673efba2a57ed99843f11f9e"><img src="https://img.shields.io/static/v1?label=Demo&message=Huggingface&color=yellow"></a> &ensp;
23
+ <a href="https://github.com/NVlabs/Sana"><img src="https://img.shields.io/static/v1?label=Code&message=Github&color=blue&logo=github"></a> &ensp;
24
+ <a href="https://nvlabs.github.io/Sana/"><img src="https://img.shields.io/static/v1?label=Project&message=Github&color=blue&logo=github-pages"></a> &ensp;
25
+ <a href="https://hanlab.mit.edu/projects/sana/"><img src="https://img.shields.io/static/v1?label=Page&message=MIT&color=darkred&logo=github-pages"></a> &ensp;
26
+ <a href="https://arxiv.org/abs/2410.10629"><img src="https://img.shields.io/static/v1?label=Arxiv&message=Sana&color=red&logo=arxiv"></a> &ensp;
27
+ <a href="https://nv-sana.mit.edu/"><img src="https://img.shields.io/static/v1?label=Demo&message=MIT&color=yellow"></a> &ensp;
28
+ <a href="https://discord.gg/rde6eaE5Ta"><img src="https://img.shields.io/static/v1?label=Discuss&message=Discord&color=purple&logo=discord"></a> &ensp;
29
+ </div>
30
+
31
+ # Model card
32
+
33
+ We incorporate a ControlNet-like(https://github.com/lllyasviel/ControlNet) module enables fine-grained control over text-to-image diffusion models.
34
+ We implement a ControlNet-Transformer architecture, specifically tailored for Transformers, achieving explicit controllability alongside high-quality image generation.
35
+
36
+
37
+ Source code is available at https://github.com/NVlabs/Sana.
38
+
39
+
40
+
41
+ <img src="https://nvlabs.github.io/Sana/asset/content/controlnet/sana_controlnet.jpg" width=640>
42
+
43
+ ## How to Use
44
+
45
+ refer to [Sana-ControlNet Guidance](https://raw.githubusercontent.com/NVlabs/Sana/refs/heads/main/asset/controlnet/controlnet_app.jpg) for more details.
46
+
47
+
48
+ ### Model Description
49
+
50
+ - **Developed by:** NVIDIA, Sana
51
+ - **Model type:** Linear-Diffusion-Transformer-based text-to-image generative model, ControlNet
52
+ - **Model size:** 900M parameters
53
+ - **Model resolution:** This model is developed to generate 1024px based images with multi-scale heigh and width.
54
+ - **License:** [NSCL v2-custom](./LICENSE.txt). Governing Terms: NVIDIA License. Additional Information: [Gemma Terms of Use | Google AI for Developers](https://ai.google.dev/gemma/terms) for Gemma-2-2B-IT, [Gemma Prohibited Use Policy | Google AI for Developers](https://ai.google.dev/gemma/prohibited_use_policy).
55
+ - **Model Description:** This is a model that can be used to generate and modify images based on text prompts.
56
+ It is a Linear Diffusion Transformer that uses one fixed, pretrained text encoders ([Gemma2-2B-IT](https://huggingface.co/google/gemma-2-2b-it))
57
+ and one 32x spatial-compressed latent feature encoder ([DC-AE](https://hanlab.mit.edu/projects/dc-ae)).
58
+ - **Special:** This model is fine-tuned from the base model [Efficient-Large-Model/Sana_600M_1024px](https://huggingface.co/Efficient-Large-Model/Sana_600M_1024px) and it supports HED ControlNet.
59
+ - **Resources for more information:** Check out our [GitHub Repository](https://github.com/NVlabs/Sana) and the [Sana report on arXiv](https://arxiv.org/abs/2410.10629).
60
+
61
+ ### Model Sources
62
+
63
+ For research purposes, we recommend our `generative-models` Github repository (https://github.com/NVlabs/Sana),
64
+ which is more suitable for both training and inference and for which most advanced diffusion sampler like Flow-DPM-Solver is integrated.
65
+ [MIT Han-Lab](https://nv-sana.mit.edu/) provides free Sana inference.
66
+ - **Repository:** ttps://github.com/NVlabs/Sana
67
+ - **Demo:** https://nv-sana.mit.edu/
68
+
69
+ ### 🧨 Diffusers
70
+
71
+ PR developing: [Sana](https://github.com/huggingface/diffusers/pull/9982) and [DC-AE](https://github.com/huggingface/diffusers/pull/9708)
72
+
73
+
74
+ ## Uses
75
+
76
+ ### Direct Use
77
+
78
+ The model is intended for research purposes only. Possible research areas and tasks include
79
+
80
+ - Generation of artworks and use in design and other artistic processes.
81
+ - Applications in educational or creative tools.
82
+ - Research on generative models.
83
+ - Safe deployment of models which have the potential to generate harmful content.
84
+
85
+ - Probing and understanding the limitations and biases of generative models.
86
+
87
+ Excluded uses are described below.
88
+
89
+ ### Out-of-Scope Use
90
+
91
+ The model was not trained to be factual or true representations of people or events, and therefore using the model to generate such content is out-of-scope for the abilities of this model.
92
+
93
+ ## Limitations and Bias
94
+
95
+ ### Limitations
96
+
97
+
98
+ - The model does not achieve perfect photorealism
99
+ - The model cannot render complex legible text
100
+ - fingers, .etc in general may not be generated properly.
101
+ - The autoencoding part of the model is lossy.
102
+
103
+ ### Bias
104
+ While the capabilities of image generation models are impressive, they can also reinforce or exacerbate social biases.