(self, image)
| 338 | return self.visual.conv1.weight.dtype |
| 339 | |
| 340 | def encode_image(self, image): |
| 341 | return self.visual(image.type(self.dtype)) |
| 342 | |
| 343 | def encode_text(self, text): |
| 344 | x = self.token_embedding(text).type(self.dtype) # [batch_size, n_ctx, d_model] |