(self, z: torch.Tensor, return_dict: bool = True)
| 829 | |
| 830 | @apply_forward_hook |
| 831 | def decode(self, z: torch.Tensor, return_dict: bool = True) -> Union[DecoderOutput, torch.Tensor]: |
| 832 | decoded = self._decode(z).sample |
| 833 | |
| 834 | if not return_dict: |
| 835 | return (decoded,) |
| 836 | return DecoderOutput(sample=decoded) |
| 837 | |
| 838 | @classmethod |
| 839 | def from_pretrained(cls, pretrained_model_path, additional_kwargs={}): |
nothing calls this directly
no test coverage detected