Add Diffusers weights

#3
by alfredplpl - opened
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": "v_prediction",
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": 96,
38
  "up_block_types": [
39
  "UpBlock2D",
40
  "CrossAttnUpBlock2D",
41
  "CrossAttnUpBlock2D",
42
  "CrossAttnUpBlock2D"
43
  ],
44
- "upcast_attention": true,
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": 768,
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",