toshas commited on
Commit
25caee5
1 Parent(s): 02ae54a

add missing fields in model_index.json

Browse files
Files changed (1) hide show
  1. model_index.json +1 -0
model_index.json CHANGED
@@ -1,6 +1,7 @@
1
  {
2
  "_class_name":"MarigoldPipeline",
3
  "_diffusers_version":"0.25.0",
 
4
  "scale_invariant": true,
5
  "shift_invariant": true,
6
  "default_denoising_steps": 1,
 
1
  {
2
  "_class_name":"MarigoldPipeline",
3
  "_diffusers_version":"0.25.0",
4
+ "prediction_type": "depth",
5
  "scale_invariant": true,
6
  "shift_invariant": true,
7
  "default_denoising_steps": 1,