(self, device: torch.device)
| 135 | return pipeline |
| 136 | |
| 137 | def to(self, device: torch.device) -> None: |
| 138 | self._device = device |
| 139 | if not self.low_vram: |
| 140 | super().to(device) |
| 141 | if self.rembg_model is not None: |
| 142 | self.rembg_model.to(device) |
| 143 | |
| 144 | def preprocess_image(self, input: Image.Image, bg_color: tuple = (0, 0, 0)) -> Image.Image: |
| 145 | """ |
no outgoing calls
no test coverage detected