| 586 | |
| 587 | @dataclass |
| 588 | class CLIPTextCfg: |
| 589 | context_length: int = 77 |
| 590 | vocab_size: int = 49408 |
| 591 | width: int = 512 |
| 592 | teacher_width: int = -1 |
| 593 | heads: int = 8 |
| 594 | layers: int = 12 |
| 595 | |
| 596 | |
| 597 | class ImageEncoder(nn.Module): |
no outgoing calls
no test coverage detected