(self, device: torch.device, **kwargs)
| 218 | |
| 219 | class FrozenImageCLIP: |
| 220 | def __init__(self, device: torch.device, **kwargs): |
| 221 | self.model = ImageCLIP(device, dtype=None, ensure_used_params=False, **kwargs) |
| 222 | for parameter in self.model.parameters(): |
| 223 | parameter.requires_grad_(False) |
| 224 | |
| 225 | @property |
| 226 | def feature_dim(self) -> int: |