Returns the names of available CLIP models
()
| 66 | |
| 67 | |
| 68 | def available_models() -> List[str]: |
| 69 | """Returns the names of available CLIP models""" |
| 70 | return list(_MODELS.keys()) |
| 71 | |
| 72 | |
| 73 | def load(name: str, device: Union[str, torch.device] = "cuda" if torch.cuda.is_available() else "cpu", jit=True): |