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),
|