Method
get_noise_pred_single
(self, latents, t, context, ref_images_pil=None, pose_cond_fea=None)
Source from the content-addressed store, hash-verified
| 24 | return next_sample |
| 25 | |
| 26 | def get_noise_pred_single(self, latents, t, context, ref_images_pil=None, pose_cond_fea=None): |
| 27 | noise_pred = self.model( |
| 28 | latents, |
| 29 | t, |
| 30 | pose_cond_fea=pose_cond_fea, |
| 31 | encoder_hidden_states=context, |
| 32 | ref_images=ref_images_pil)["sample"] |
| 33 | return noise_pred |
| 34 | |
| 35 | @torch.no_grad() |
| 36 | def init_emb_img(self, clip_emb_im=None): |
Tested by
no test coverage detected