| 949 | |
| 950 | @register_to_config |
| 951 | def __init__( |
| 952 | self, |
| 953 | do_resize: bool = True, |
| 954 | vae_scale_factor: int = 8, |
| 955 | resample: str = "lanczos", |
| 956 | do_normalize: bool = False, |
| 957 | do_binarize: bool = True, |
| 958 | do_convert_grayscale: bool = True, |
| 959 | ): |
| 960 | super().__init__( |
| 961 | do_resize=do_resize, |
| 962 | vae_scale_factor=vae_scale_factor, |
| 963 | resample=resample, |
| 964 | do_normalize=do_normalize, |
| 965 | do_binarize=do_binarize, |
| 966 | do_convert_grayscale=do_convert_grayscale, |
| 967 | ) |
| 968 | |
| 969 | @staticmethod |
| 970 | def downsample(mask: torch.Tensor, batch_size: int, num_queries: int, value_embed_dim: int): |