Update config.json
Browse files- config.json +3 -3
config.json
CHANGED
|
@@ -36,7 +36,7 @@
|
|
| 36 |
"device": "cuda",
|
| 37 |
"use_amp": false,
|
| 38 |
"push_to_hub": true,
|
| 39 |
-
"repo_id": "Polarcocola/
|
| 40 |
"private": null,
|
| 41 |
"tags": null,
|
| 42 |
"license": null,
|
|
@@ -68,7 +68,7 @@
|
|
| 68 |
"n_groups": 8,
|
| 69 |
"diffusion_step_embed_dim": 128,
|
| 70 |
"use_film_scale_modulation": true,
|
| 71 |
-
"noise_scheduler_type": "
|
| 72 |
"num_train_timesteps": 100,
|
| 73 |
"beta_schedule": "squaredcos_cap_v2",
|
| 74 |
"beta_start": 0.0001,
|
|
@@ -76,7 +76,7 @@
|
|
| 76 |
"prediction_type": "epsilon",
|
| 77 |
"clip_sample": true,
|
| 78 |
"clip_sample_range": 1.0,
|
| 79 |
-
"num_inference_steps":
|
| 80 |
"do_mask_loss_for_padding": false,
|
| 81 |
"optimizer_lr": 0.0001,
|
| 82 |
"optimizer_betas": [
|
|
|
|
| 36 |
"device": "cuda",
|
| 37 |
"use_amp": false,
|
| 38 |
"push_to_hub": true,
|
| 39 |
+
"repo_id": "Polarcocola/model_2025_11_16_4_diffusion_10w",
|
| 40 |
"private": null,
|
| 41 |
"tags": null,
|
| 42 |
"license": null,
|
|
|
|
| 68 |
"n_groups": 8,
|
| 69 |
"diffusion_step_embed_dim": 128,
|
| 70 |
"use_film_scale_modulation": true,
|
| 71 |
+
"noise_scheduler_type": "DDIM",
|
| 72 |
"num_train_timesteps": 100,
|
| 73 |
"beta_schedule": "squaredcos_cap_v2",
|
| 74 |
"beta_start": 0.0001,
|
|
|
|
| 76 |
"prediction_type": "epsilon",
|
| 77 |
"clip_sample": true,
|
| 78 |
"clip_sample_range": 1.0,
|
| 79 |
+
"num_inference_steps": 10,
|
| 80 |
"do_mask_loss_for_padding": false,
|
| 81 |
"optimizer_lr": 0.0001,
|
| 82 |
"optimizer_betas": [
|