Polarcocola commited on
Commit
5276f9a
·
verified ·
1 Parent(s): b0bdd33

Update config.json

Browse files
Files changed (1) hide show
  1. 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/model_2025_11_16_4_act_10w",
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": "DDPM",
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": null,
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": [