Upload 4 files
Browse filesUpdate CM EDM test class unconditional U-Net checkpoint.
- model_index.json +1 -1
- scheduler/scheduler_config.json +2 -2
- unet/config.json +2 -1
model_index.json
CHANGED
|
@@ -1,6 +1,6 @@
|
|
| 1 |
{
|
| 2 |
"_class_name": "KarrasEDMPipeline",
|
| 3 |
-
"_diffusers_version": "0.
|
| 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.
|
| 4 |
"clip_sample": false,
|
| 5 |
"clip_sample_range": 1.0,
|
| 6 |
"num_train_timesteps": 40,
|
| 7 |
"precondition_type": "edm",
|
| 8 |
-
"prediction_type": "
|
| 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.
|
| 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": null,
|
|
|
|
| 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": null,
|
| 29 |
+
"num_train_timesteps": null,
|
| 30 |
"out_channels": 3,
|
| 31 |
"resnet_time_scale_shift": "scale_shift",
|
| 32 |
"sample_size": 32,
|