patrickvonplaten commited on
Commit
dc431a9
·
1 Parent(s): 27a6be0
Files changed (4) hide show
  1. config.json +43 -0
  2. diffusion_model.pt +3 -0
  3. model_index.json +12 -0
  4. scheduler_config.json +7 -0
config.json ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "NCSNpp",
3
+ "_diffusers_version": "0.0.4",
4
+ "attention_type": "ddpm",
5
+ "attn_resolutions": [
6
+ 16
7
+ ],
8
+ "centered": true,
9
+ "ch_mult": [
10
+ 1,
11
+ 2,
12
+ 2,
13
+ 2
14
+ ],
15
+ "conditional": true,
16
+ "continuous": true,
17
+ "conv_size": 3,
18
+ "dropout": 0.0,
19
+ "embedding_type": "positional",
20
+ "fir": false,
21
+ "fir_kernel": [
22
+ 1,
23
+ 3,
24
+ 3,
25
+ 1
26
+ ],
27
+ "fourier_scale": 16,
28
+ "image_size": 32,
29
+ "init_scale": 0.0,
30
+ "name_or_path": "./",
31
+ "nf": 128,
32
+ "nonlinearity": "swish",
33
+ "normalization": "GroupNorm",
34
+ "num_channels": 3,
35
+ "num_res_blocks": 8,
36
+ "progressive": "none",
37
+ "progressive_combine": "sum",
38
+ "progressive_input": "none",
39
+ "resamp_with_conv": true,
40
+ "resblock_type": "biggan",
41
+ "scale_by_sigma": false,
42
+ "skip_rescale": true
43
+ }
diffusion_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4030d46de2576bd20ad45d52e3262bae570eafa0384ddd99baf8f5e4a514a95a
3
+ size 426855303
model_index.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ScoreSdeVpPipeline",
3
+ "_diffusers_version": "0.0.4",
4
+ "model": [
5
+ "diffusers",
6
+ "NCSNpp"
7
+ ],
8
+ "scheduler": [
9
+ "diffusers",
10
+ "ScoreSdeVpScheduler"
11
+ ]
12
+ }
scheduler_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_class_name": "ScoreSdeVpScheduler",
3
+ "_diffusers_version": "0.0.4",
4
+ "beta_max": 20,
5
+ "beta_min": 0.1,
6
+ "sampling_eps": 0.001
7
+ }