(self, path, dropout_prob=0.1)
| 64 | Embeds text prompt into vector representations. Also handles text dropout for classifier-free guidance. |
| 65 | """ |
| 66 | def __init__(self, path, dropout_prob=0.1): |
| 67 | super().__init__() |
| 68 | self.text_encodder = FrozenCLIPEmbedder(path=path) |
| 69 | self.dropout_prob = dropout_prob |
| 70 | |
| 71 | def token_drop(self, text_prompts, force_drop_ids=None): |
| 72 | """ |
no test coverage detected