coralLight commited on
Commit
4146ea9
·
1 Parent(s): 689fe99

add inference

Browse files
Files changed (1) hide show
  1. uni_pc.py +1 -1
uni_pc.py CHANGED
@@ -549,7 +549,7 @@ class UniPC:
549
  if not self.use_afs:
550
  fir_output = self.noise_prediction_fn(x, vec_t)
551
  else:
552
- fir_output = x * 0.98 # ultilize npnet there in the future
553
  if npnet_x is not None and npnet_scale is not None:
554
  fir_output = npnet_x
555
  # fir_output = fir_output - npnet_scale * (npnet_out - fir_output) #guidance_scale * (noise - noise_uncond)
 
549
  if not self.use_afs:
550
  fir_output = self.noise_prediction_fn(x, vec_t)
551
  else:
552
+ fir_output = x # ultilize npnet there in the future
553
  if npnet_x is not None and npnet_scale is not None:
554
  fir_output = npnet_x
555
  # fir_output = fir_output - npnet_scale * (npnet_out - fir_output) #guidance_scale * (noise - noise_uncond)