(self)
| 152 | |
| 153 | @property |
| 154 | def dtype(self): |
| 155 | try: |
| 156 | return self.visual.conv1.weight.dtype |
| 157 | except: |
| 158 | try: |
| 159 | return self.visual.head.weight.dtype |
| 160 | except: |
| 161 | try: |
| 162 | return self.visual.stem[0].weight.dtype |
| 163 | except: |
| 164 | return self.text_encoder.text_projection.weight.dtype |
| 165 | |
| 166 | def encode_image(self, image): |
| 167 | return self.visual(image.type(self.dtype)) |