Sets the model to evaluation mode. Overrides the default eval() method to also set the GPT model to eval mode.
(self)
| 706 | return Xtts(config) |
| 707 | |
| 708 | def eval(self): # pylint: disable=redefined-builtin |
| 709 | """Sets the model to evaluation mode. Overrides the default eval() method to also set the GPT model to eval mode.""" |
| 710 | self.gpt.init_gpt_for_inference() |
| 711 | super().eval() |
| 712 | |
| 713 | def get_compatible_checkpoint_state_dict(self, model_path): |
| 714 | checkpoint = load_fsspec(model_path, map_location=torch.device("cpu"))["model"] |