diff --git a/generative/inferers/inferer.py b/generative/inferers/inferer.py index 5d06b12a..40c3459b 100644 --- a/generative/inferers/inferer.py +++ b/generative/inferers/inferer.py @@ -69,7 +69,7 @@ def sample( save_intermediates: bool | None = False, intermediate_steps: int | None = 100, conditioning: torch.Tensor | None = None, - verbose: bool | None = True, + verbose: bool = True, ) -> torch.Tensor | tuple[torch.Tensor, list[torch.Tensor]]: """ Args: @@ -114,7 +114,7 @@ def get_likelihood( conditioning: torch.Tensor | None = None, original_input_range: tuple | None = (0, 255), scaled_input_range: tuple | None = (0, 1), - verbose: bool | None = True, + verbose: bool = True, ) -> torch.Tensor | tuple[torch.Tensor, list[torch.Tensor]]: """ Computes the log-likelihoods for an input. @@ -321,7 +321,7 @@ def sample( save_intermediates: bool | None = False, intermediate_steps: int | None = 100, conditioning: torch.Tensor | None = None, - verbose: bool | None = True, + verbose: bool = True, ) -> torch.Tensor | tuple[torch.Tensor, list[torch.Tensor]]: """ Args: @@ -371,7 +371,7 @@ def get_likelihood( conditioning: torch.Tensor | None = None, original_input_range: tuple | None = (0, 255), scaled_input_range: tuple | None = (0, 1), - verbose: bool | None = True, + verbose: bool = True, resample_latent_likelihoods: bool = False, resample_interpolation_mode: str = "nearest", ) -> torch.Tensor | tuple[torch.Tensor, list[torch.Tensor]]: @@ -471,7 +471,7 @@ def sample( conditioning: torch.Tensor | None = None, temperature: float = 1.0, top_k: int | None = None, - verbose: bool | None = True, + verbose: bool = True, ) -> torch.Tensor: """ Sampling function for the VQVAE + Transformer model.