Swap checkpoint_dir, checkpoint_path parameters in xtts

TTS/tts/models/xtts.py

@@ -731,8 +731,8 @@ def get_compatible_checkpoint_state_dict(self, model_path):
  def load_checkpoint(
  self,
  config,
- checkpoint_dir=None,
  checkpoint_path=None,
+ checkpoint_dir=None,
  vocab_path=None,
  eval=True,
  strict=True,
@@ -744,16 +744,17 @@ def load_checkpoint(
 
  Args:
  config (dict): The configuration dictionary for the model.
- checkpoint_dir (str, optional): The directory where the checkpoint is stored. Defaults to None.
  checkpoint_path (str, optional): The path to the checkpoint file. Defaults to None.
+ checkpoint_dir (str, optional): The directory where the checkpoint is stored. Defaults to None.
  vocab_path (str, optional): The path to the vocabulary file. Defaults to None.
  eval (bool, optional): Whether to set the model to evaluation mode. Defaults to True.
  strict (bool, optional): Whether to strictly enforce that the keys in the checkpoint match the keys in the model. Defaults to True.
 
  Returns:
  None
  """
-
+ if checkpoint_dir is None and checkpoint_path:
+ checkpoint_dir = os.path.dirname(checkpoint_path)
  model_path = checkpoint_path or os.path.join(checkpoint_dir, "model.pth")
  vocab_path = vocab_path or os.path.join(checkpoint_dir, "vocab.json")