Ameyapores
/

push_block_dp_aug_8_2025_staticimg

Model card Files Files and versions

Ameyapores commited on 9 days ago

Commit

e952520

·

verified ·

1 Parent(s): 880b9da

adding the num_inference_steps

Files changed (1) hide show

train_config.json +1 -1

train_config.json CHANGED Viewed

@@ -134,7 +134,7 @@
         "prediction_type": "epsilon",
         "clip_sample": true,
         "clip_sample_range": 1.0,
-        "num_inference_steps": null,
         "do_mask_loss_for_padding": false,
         "optimizer_lr": 0.0001,
         "optimizer_betas": [

         "prediction_type": "epsilon",
         "clip_sample": true,
         "clip_sample_range": 1.0,
+        "num_inference_steps": 10,
         "do_mask_loss_for_padding": false,
         "optimizer_lr": 0.0001,
         "optimizer_betas": [