(
self,
device: Optional[torch.device | str] = None,
dtype: Optional[torch.dtype | str] = None,
)
| 46 | return model |
| 47 | |
| 48 | def processor( |
| 49 | self, |
| 50 | device: Optional[torch.device | str] = None, |
| 51 | dtype: Optional[torch.dtype | str] = None, |
| 52 | ) -> SegformerImageProcessor: |
| 53 | return SegformerImageProcessor.from_pretrained(self.checkpoint) |
no test coverage detected