Ubuntu commited on
Commit
96141cc
1 Parent(s): 4acdd6d

diffusers support

Browse files
model_index.json CHANGED
@@ -12,7 +12,7 @@
12
  ],
13
  "scheduler": [
14
  "diffusers",
15
- "EulerDiscreteScheduler"
16
  ],
17
  "text_encoder": [
18
  "transformers",
 
12
  ],
13
  "scheduler": [
14
  "diffusers",
15
+ "EDMEulerScheduler"
16
  ],
17
  "text_encoder": [
18
  "transformers",
scheduler/scheduler_config.json CHANGED
@@ -1,22 +1,10 @@
1
  {
2
- "_class_name": "EulerDiscreteScheduler",
3
  "_diffusers_version": "0.27.0.dev0",
4
- "beta_end": 0.012,
5
- "beta_schedule": "scaled_linear",
6
- "beta_start": 0.00085,
7
- "clip_sample": false,
8
- "interpolation_type": "linear",
9
  "num_train_timesteps": 1000,
10
  "prediction_type": "epsilon",
11
- "rescale_betas_zero_snr": false,
12
- "sample_max_value": 1.0,
13
- "set_alpha_to_one": false,
14
- "sigma_max": null,
15
- "sigma_min": null,
16
- "skip_prk_steps": true,
17
- "steps_offset": 1,
18
- "timestep_spacing": "leading",
19
- "timestep_type": "discrete",
20
- "trained_betas": null,
21
- "use_karras_sigmas": false
22
  }
 
1
  {
2
+ "_class_name": "EDMEulerScheduler",
3
  "_diffusers_version": "0.27.0.dev0",
 
 
 
 
 
4
  "num_train_timesteps": 1000,
5
  "prediction_type": "epsilon",
6
+ "rho": 7.0,
7
+ "sigma_data": 0.5,
8
+ "sigma_max": 80.0,
9
+ "sigma_min": 0.002
 
 
 
 
 
 
 
10
  }
vae/config.json CHANGED
@@ -21,24 +21,23 @@
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
  "sample_size": 1024,
24
- "scaling_factor": 0.13025,
25
  "up_block_types": [
26
  "UpDecoderBlock2D",
27
  "UpDecoderBlock2D",
28
  "UpDecoderBlock2D",
29
  "UpDecoderBlock2D"
30
  ],
31
- "edm_mean": [
32
  -1.6574,
33
  1.886,
34
  -1.383,
35
  2.5155
36
  ],
37
- "edm_std": [
38
  8.4927,
39
  5.9022,
40
  6.5498,
41
  5.2299
42
  ],
43
- "edm_scale": 0.5
44
  }
 
21
  "norm_num_groups": 32,
22
  "out_channels": 3,
23
  "sample_size": 1024,
 
24
  "up_block_types": [
25
  "UpDecoderBlock2D",
26
  "UpDecoderBlock2D",
27
  "UpDecoderBlock2D",
28
  "UpDecoderBlock2D"
29
  ],
30
+ "latents_mean": [
31
  -1.6574,
32
  1.886,
33
  -1.383,
34
  2.5155
35
  ],
36
+ "latents_std": [
37
  8.4927,
38
  5.9022,
39
  6.5498,
40
  5.2299
41
  ],
42
+ "scaling_factor": 0.5
43
  }