Method __init__

diffbir/utils/caption.py:42–43

(self, device: torch.device)

class Captioner:

    def __init__(self, device: torch.device) -> "Captioner":
        self.device = device

    @overload
    def __call__(self, image: Image.Image) -> str: ...