Update modeling_vits.py

modeling_vits.py

@@ -104,7 +104,7 @@ class VitsModel(VitsPreTrainedModel):
         else:
             input_padding_mask = torch.ones_like(input_ids).unsqueeze(-1).to(mask_dtype)
 
-        if self.config.num_speakers > 1 and speaker_id is not None:
+        if self.config.num_speakers > 1 and speaker_id is not None and self.config.num_emotions > 1 and emotion_id is not None:
             if not 0 <= speaker_id < self.config.num_speakers:
                 raise ValueError(f"Set `speaker_id` in the range 0-{self.config.num_speakers - 1}. or Set `emotion_id` in the range 0-{self.config.num_emotions - 1}.")
             if isinstance(speaker_id, int) and isinstance(emotion_id, int):