drscotthawley commited on
Commit
5224929
1 Parent(s): bf69a1f

turing off chord_cond for demo

Browse files
Files changed (1) hide show
  1. sample.py +4 -1
sample.py CHANGED
@@ -600,9 +600,12 @@ def main():
600
  #print("init_data.shape, init_data.min, init_data.max = ", init_data.shape, init_data.min(), init_data.max())
601
  else:
602
  init_data, mask, chord_cond = None, None, None
 
 
603
 
604
  print("chord_cond = ", chord_cond)
605
- extra_args['chord_cond'] = chord_cond
 
606
  # these two work:
607
  #x_0 = K.sampling.sample_lms(model, x, sigmas, disable=not accelerator.is_local_main_process, extra_args=extra_args)
608
  #x_0 = K.sampling.sample_dpmpp_2m_sde(model, x, sigmas, disable=not accelerator.is_local_main_process, extra_args=extra_args)
 
600
  #print("init_data.shape, init_data.min, init_data.max = ", init_data.shape, init_data.min(), init_data.max())
601
  else:
602
  init_data, mask, chord_cond = None, None, None
603
+ # chord_cond doesn't work anyway so f it:
604
+ chord_cond = None
605
 
606
  print("chord_cond = ", chord_cond)
607
+ if chord_cond is not None:
608
+ extra_args['chord_cond'] = chord_cond
609
  # these two work:
610
  #x_0 = K.sampling.sample_lms(model, x, sigmas, disable=not accelerator.is_local_main_process, extra_args=extra_args)
611
  #x_0 = K.sampling.sample_dpmpp_2m_sde(model, x, sigmas, disable=not accelerator.is_local_main_process, extra_args=extra_args)