weifeng-chen commited on
Commit
6939f65
1 Parent(s): 9c042a9
Files changed (3) hide show
  1. README.md +1 -0
  2. config.json +36 -0
  3. pytorch_model.bin +3 -0
README.md ADDED
@@ -0,0 +1 @@
 
 
1
+ This model is finetune from Katherine Crowson's fine-tuned 512x512 diffusion model (https://github.com/openai/guided-diffusion), using 1k+ nature image crawled from the InterNet.
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:751eab02340896cd965aeefa3a6d6532fea6a2be3f28ff5c80de881404499401
3
+ size 2232098937