(self, clip_version)
| 87 | self.poseFinal = nn.Linear(self.latent_dim, self.input_feats) |
| 88 | |
| 89 | def load_and_freeze_clip(self, clip_version): |
| 90 | clip_model, _ = clip.load(clip_version, device='cpu', jit=False) |
| 91 | clip.model.convert_weights(clip_model) |
| 92 | |
| 93 | clip_model.eval() |
| 94 | for p in clip_model.parameters(): |
| 95 | p.requires_grad = False |
| 96 | |
| 97 | return clip_model |
| 98 | |
| 99 | def mask_cond(self, cond, force_mask=False): |
| 100 | bs = cond.shape[0] |