weifeng-chen
commited on
Commit
•
53576b9
1
Parent(s):
fd22b1d
cyberpunk init
Browse files- README.md +63 -0
- config.json +36 -0
- pytorch_model.bin +3 -0
README.md
ADDED
@@ -0,0 +1,63 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
# inference: false
|
4 |
+
# pipeline_tag: zero-shot-image-classification
|
5 |
+
pipeline_tag: feature-extraction
|
6 |
+
|
7 |
+
# inference:
|
8 |
+
# parameters:
|
9 |
+
tags:
|
10 |
+
- text-to-image
|
11 |
+
---
|
12 |
+
|
13 |
+
# Taiyi-Diffusion-532M-Nature
|
14 |
+
|
15 |
+
- Github: [Fengshenbang-LM](https://github.com/IDEA-CCNL/Fengshenbang-LM)
|
16 |
+
- Docs: [Fengshenbang-Docs](https://fengshenbang-doc.readthedocs.io/)
|
17 |
+
|
18 |
+
## 简介 Brief Introduction
|
19 |
+
|
20 |
+
该模型由[Katherine Crowson's](https://github.com/openai/guided-diffusion)的无条件扩散模型在1k+张收集的赛博朋克风的图上微调而来。
|
21 |
+
|
22 |
+
This model is finetune from Katherine Crowson's fine-tuned 512x512 diffusion model (https://github.com/openai/guided-diffusion), using 1k+ cyberpunk-style image crawled from the InterNet.
|
23 |
+
|
24 |
+
## 模型分类 Model Taxonomy
|
25 |
+
|
26 |
+
| 需求 Demand | 任务 Task | 系列 Series | 模型 Model | 参数 Parameter | 额外 Extra |
|
27 |
+
| :----: | :----: | :----: | :----: | :----: | :----: |
|
28 |
+
| 特殊 Special | 多模态 Multimodal | 太乙 Taiyi | Diffusion Model | 532M | Cyberpunk |
|
29 |
+
|
30 |
+
## 使用 Usage
|
31 |
+
|
32 |
+
使用示例见:https://github.com/IDEA-CCNL/Fengshenbang-LM/tree/dev_cwf/fengshen/examples/disco_project
|
33 |
+
|
34 |
+
|
35 |
+
## 引用 Citation
|
36 |
+
|
37 |
+
如果您在您的工作中使用了我们的模型,可以引用我们的[论文](https://arxiv.org/abs/2209.02970):
|
38 |
+
|
39 |
+
If you are using the resource for your work, please cite the our [paper](https://arxiv.org/abs/2209.02970):
|
40 |
+
|
41 |
+
```text
|
42 |
+
@article{fengshenbang,
|
43 |
+
author = {Junjie Wang and Yuxiang Zhang and Lin Zhang and Ping Yang and Xinyu Gao and Ziwei Wu and Xiaoqun Dong and Junqing He and Jianheng Zhuo and Qi Yang and Yongfeng Huang and Xiayu Li and Yanghan Wu and Junyu Lu and Xinyu Zhu and Weifeng Chen and Ting Han and Kunhao Pan and Rui Wang and Hao Wang and Xiaojun Wu and Zhongshen Zeng and Chongpei Chen and Ruyi Gan and Jiaxing Zhang},
|
44 |
+
title = {Fengshenbang 1.0: Being the Foundation of Chinese Cognitive Intelligence},
|
45 |
+
journal = {CoRR},
|
46 |
+
volume = {abs/2209.02970},
|
47 |
+
year = {2022}
|
48 |
+
}
|
49 |
+
```
|
50 |
+
|
51 |
+
也可以引用我们的[网站](https://github.com/IDEA-CCNL/Fengshenbang-LM/):
|
52 |
+
|
53 |
+
You can also cite our [website](https://github.com/IDEA-CCNL/Fengshenbang-LM/):
|
54 |
+
|
55 |
+
```text
|
56 |
+
@misc{Fengshenbang-LM,
|
57 |
+
title={Fengshenbang-LM},
|
58 |
+
author={IDEA-CCNL},
|
59 |
+
year={2021},
|
60 |
+
howpublished={\url{https://github.com/IDEA-CCNL/Fengshenbang-LM}},
|
61 |
+
}
|
62 |
+
```
|
63 |
+
|
config.json
ADDED
@@ -0,0 +1,36 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"architectures": [
|
3 |
+
"HFUNetModel"
|
4 |
+
],
|
5 |
+
"attention_resolutions": [
|
6 |
+
16,
|
7 |
+
32,
|
8 |
+
64
|
9 |
+
],
|
10 |
+
"channel_mult": [
|
11 |
+
0.5,
|
12 |
+
1,
|
13 |
+
1,
|
14 |
+
2,
|
15 |
+
2,
|
16 |
+
4,
|
17 |
+
4
|
18 |
+
],
|
19 |
+
"dropout": 0.0,
|
20 |
+
"image_size": 512,
|
21 |
+
"in_channels": 3,
|
22 |
+
"model_channels": 256,
|
23 |
+
"num_classes": null,
|
24 |
+
"num_head_channels": 64,
|
25 |
+
"num_heads": 4,
|
26 |
+
"num_heads_upsample": -1,
|
27 |
+
"num_res_blocks": 2,
|
28 |
+
"out_channels": 6,
|
29 |
+
"resblock_updown": true,
|
30 |
+
"torch_dtype": "float32",
|
31 |
+
"transformers_version": "4.22.2",
|
32 |
+
"use_checkpoint": false,
|
33 |
+
"use_fp16": true,
|
34 |
+
"use_new_attention_order": false,
|
35 |
+
"use_scale_shift_norm": true
|
36 |
+
}
|
pytorch_model.bin
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:e2d7a64c0173d69bfaa462c6421a0f919e046ce49fd6b83b4df0d4d0bcc24a0f
|
3 |
+
size 2232098937
|