(self, model: MDMModel, num_tokens: int)
| 184 | |
| 185 | class ExportableLingBotDepth(nn.Module): |
| 186 | def __init__(self, model: MDMModel, num_tokens: int): |
| 187 | super().__init__() |
| 188 | self.model = model |
| 189 | self.num_tokens = int(num_tokens) |
| 190 | |
| 191 | def forward(self, image: torch.Tensor, depth: torch.Tensor) -> torch.Tensor: |
| 192 | output = self.model.forward( |
nothing calls this directly
no outgoing calls
no test coverage detected