lopho commited on
Commit
63d774d
1 Parent(s): 8d0d1af

use image cfg for hint image gen

Browse files
Files changed (1) hide show
  1. makeavid_sd/inference.py +1 -1
makeavid_sd/inference.py CHANGED
@@ -370,7 +370,7 @@ class InferenceUNetPseudo3D:
370
  tt,
371
  encoder_hidden_states = encoded_neg_prompt
372
  ).sample
373
- noise_pred = noise_pred_uncond + cfg * (noise_pred - noise_pred_uncond)
374
  image_latents, image_scheduler_state = scheduler.step(
375
  image_scheduler_state,
376
  noise_pred.astype(jnp.float32),
 
370
  tt,
371
  encoder_hidden_states = encoded_neg_prompt
372
  ).sample
373
+ noise_pred = noise_pred_uncond + cfg_image * (noise_pred - noise_pred_uncond)
374
  image_latents, image_scheduler_state = scheduler.step(
375
  image_scheduler_state,
376
  noise_pred.astype(jnp.float32),