Add Diffusers weights
#3
by
alfredplpl
- opened
- scheduler/scheduler_config.json +1 -1
- unet/config.json +2 -2
- vae/config.json +1 -1
scheduler/scheduler_config.json
CHANGED
@@ -6,7 +6,7 @@
|
|
6 |
"beta_start": 0.00085,
|
7 |
"clip_sample": false,
|
8 |
"num_train_timesteps": 1000,
|
9 |
-
"prediction_type": "
|
10 |
"set_alpha_to_one": false,
|
11 |
"skip_prk_steps": true,
|
12 |
"steps_offset": 1,
|
|
|
6 |
"beta_start": 0.00085,
|
7 |
"clip_sample": false,
|
8 |
"num_train_timesteps": 1000,
|
9 |
+
"prediction_type": "epsilon",
|
10 |
"set_alpha_to_one": false,
|
11 |
"skip_prk_steps": true,
|
12 |
"steps_offset": 1,
|
unet/config.json
CHANGED
@@ -34,13 +34,13 @@
|
|
34 |
"num_class_embeds": null,
|
35 |
"only_cross_attention": false,
|
36 |
"out_channels": 4,
|
37 |
-
"sample_size":
|
38 |
"up_block_types": [
|
39 |
"UpBlock2D",
|
40 |
"CrossAttnUpBlock2D",
|
41 |
"CrossAttnUpBlock2D",
|
42 |
"CrossAttnUpBlock2D"
|
43 |
],
|
44 |
-
"upcast_attention":
|
45 |
"use_linear_projection": true
|
46 |
}
|
|
|
34 |
"num_class_embeds": null,
|
35 |
"only_cross_attention": false,
|
36 |
"out_channels": 4,
|
37 |
+
"sample_size": 64,
|
38 |
"up_block_types": [
|
39 |
"UpBlock2D",
|
40 |
"CrossAttnUpBlock2D",
|
41 |
"CrossAttnUpBlock2D",
|
42 |
"CrossAttnUpBlock2D"
|
43 |
],
|
44 |
+
"upcast_attention": false,
|
45 |
"use_linear_projection": true
|
46 |
}
|
vae/config.json
CHANGED
@@ -19,7 +19,7 @@
|
|
19 |
"layers_per_block": 2,
|
20 |
"norm_num_groups": 32,
|
21 |
"out_channels": 3,
|
22 |
-
"sample_size":
|
23 |
"up_block_types": [
|
24 |
"UpDecoderBlock2D",
|
25 |
"UpDecoderBlock2D",
|
|
|
19 |
"layers_per_block": 2,
|
20 |
"norm_num_groups": 32,
|
21 |
"out_channels": 3,
|
22 |
+
"sample_size": 512,
|
23 |
"up_block_types": [
|
24 |
"UpDecoderBlock2D",
|
25 |
"UpDecoderBlock2D",
|