| 27 | |
| 28 | @dataclass |
| 29 | class OpenCLIPNetworkConfig: |
| 30 | _target: Type = field(default_factory=lambda: OpenCLIPNetwork) |
| 31 | clip_model_type: str = "ViT-B-16" |
| 32 | clip_model_pretrained: str = "laion2b_s34b_b88k" |
| 33 | clip_n_dims: int = 512 |
| 34 | negatives: Tuple[str] = ("object", "things", "stuff", "texture") |
| 35 | positives: Tuple[str] = ("",) |
| 36 | |
| 37 | class OpenCLIPNetwork(nn.Module): |
| 38 | def __init__(self, config: OpenCLIPNetworkConfig): |
nothing calls this directly
no outgoing calls
no test coverage detected