Hugo Flores Garcia commited on
Commit
62f49b0
1 Parent(s): 46c305c

pass sampling steps to coarse2fine

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -126,6 +126,7 @@ def _vamp(data, return_mask=False):
126
  mask_temperature=data[masktemp]*10,
127
  sampling_temperature=data[sampletemp],
128
  mask=mask,
 
129
  sample_cutoff=data[sample_cutoff],
130
  seed=_seed,
131
  )
 
126
  mask_temperature=data[masktemp]*10,
127
  sampling_temperature=data[sampletemp],
128
  mask=mask,
129
+ sampling_steps=data[num_steps],
130
  sample_cutoff=data[sample_cutoff],
131
  seed=_seed,
132
  )