trytoycon
/

diffusion_pick_place_60_30ksteps_phase_a

Model card Files Files and versions

Avinash711 commited on 11 days ago

Commit

629bda8

·

verified ·

1 Parent(s): d915f8c

Update config.json

Files changed (1) hide show

config.json +1 -1

config.json CHANGED Viewed

@@ -76,7 +76,7 @@
     "prediction_type": "epsilon",
     "clip_sample": true,
     "clip_sample_range": 1.0,
-    "num_inference_steps": 10,
     "compile_model": false,
     "compile_mode": "reduce-overhead",
     "do_mask_loss_for_padding": false,

     "prediction_type": "epsilon",
     "clip_sample": true,
     "clip_sample_range": 1.0,
+    "num_inference_steps": 18,
     "compile_model": false,
     "compile_mode": "reduce-overhead",
     "do_mask_loss_for_padding": false,