Hugo Flores Garcia commited on
Commit
88c78e1
1 Parent(s): 07aa55e

num steps should propagate to coarse2fine too

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -114,7 +114,12 @@ def _vamp(data, return_mask=False):
114
  )
115
 
116
  if use_coarse2fine:
117
- zv = interface.coarse_to_fine(zv, temperature=data[temp], mask=mask)
 
 
 
 
 
118
 
119
  sig = interface.to_signal(zv).cpu()
120
  print("done")
 
114
  )
115
 
116
  if use_coarse2fine:
117
+ zv = interface.coarse_to_fine(
118
+ zv,
119
+ temperature=data[temp],
120
+ mask=mask,
121
+ sampling_steps=data[num_steps]
122
+ )
123
 
124
  sig = interface.to_signal(zv).cpu()
125
  print("done")