in src/image_gen_aux/preprocessors/teed/teed_preprocessor.py [0:0]
def __call__(
self,
image: Union[PIL.Image.Image, np.ndarray, torch.Tensor, List[PIL.Image.Image]],
resolution_scale: float = 1.0,
invert: bool = False,
safe_steps: int = 2,
batch_size: int = 1,
return_type: str = "pil",