Spaces:
Runtime error
Runtime error
use image cfg for hint image gen
Browse files- 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 +
|
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),
|