dg845 commited on
Commit
cfba282
1 Parent(s): 32b7e0b

Upload 4 files

Browse files

Update CM EDM test class conditional U-Net checkpoint.

model_index.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "KarrasEDMPipeline",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "scheduler": [
5
  "diffusers",
6
  "KarrasEDMScheduler"
 
1
  {
2
  "_class_name": "KarrasEDMPipeline",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "scheduler": [
5
  "diffusers",
6
  "KarrasEDMScheduler"
scheduler/scheduler_config.json CHANGED
@@ -1,11 +1,11 @@
1
  {
2
  "_class_name": "KarrasEDMScheduler",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "clip_sample": false,
5
  "clip_sample_range": 1.0,
6
  "num_train_timesteps": 40,
7
  "precondition_type": "edm",
8
- "prediction_type": "edm",
9
  "rho": 7.0,
10
  "s_churn": 0.0,
11
  "s_noise": 1.0,
 
1
  {
2
  "_class_name": "KarrasEDMScheduler",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "clip_sample": false,
5
  "clip_sample_range": 1.0,
6
  "num_train_timesteps": 40,
7
  "precondition_type": "edm",
8
+ "prediction_type": "sample",
9
  "rho": 7.0,
10
  "s_churn": 0.0,
11
  "s_noise": 1.0,
unet/config.json CHANGED
@@ -1,6 +1,6 @@
1
  {
2
  "_class_name": "UNet2DModel",
3
- "_diffusers_version": "0.22.0.dev0",
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 8,
@@ -26,6 +26,7 @@
26
  "norm_eps": 1e-05,
27
  "norm_num_groups": 32,
28
  "num_class_embeds": 1000,
 
29
  "out_channels": 3,
30
  "resnet_time_scale_shift": "scale_shift",
31
  "sample_size": 32,
 
1
  {
2
  "_class_name": "UNet2DModel",
3
+ "_diffusers_version": "0.24.0.dev0",
4
  "act_fn": "silu",
5
  "add_attention": true,
6
  "attention_head_dim": 8,
 
26
  "norm_eps": 1e-05,
27
  "norm_num_groups": 32,
28
  "num_class_embeds": 1000,
29
+ "num_train_timesteps": null,
30
  "out_channels": 3,
31
  "resnet_time_scale_shift": "scale_shift",
32
  "sample_size": 32,