Cqy2019 MCplayer commited on
Commit
5cfff43
verified
1 Parent(s): 6957f7a

update_readme (#1)

Browse files

- update readme (dc16cdd883bba3fbef64c793b0b5c04bc1f6f681)


Co-authored-by: 甯告垚 <[email protected]>

Files changed (2) hide show
  1. README.md +2 -2
  2. modeling_moss_ttsd.py +1 -1
README.md CHANGED
@@ -30,8 +30,8 @@ import os
30
  import torchaudio
31
  from transformers import AutoModel, AutoProcessor
32
 
33
- processor = AutoProcessor.from_pretrained("fnlp/MOSS-TTSD-v0.7", codec_path="fnlp/XY_Tokenizer_TTSD_V0_hf", trust_remote_code=True)
34
- model = AutoModel.from_pretrained("fnlp/MOSS-TTSD-v0.7", trust_remote_code=True, device_map="auto").eval()
35
 
36
  data = [{
37
  "base_path": "/path/to/audio/files",
 
30
  import torchaudio
31
  from transformers import AutoModel, AutoProcessor
32
 
33
+ processor = AutoProcessor.from_pretrained("OpenMOSS-Team/MOSS-TTSD-v0.7", codec_path="OpenMOSS-Team/XY_Tokenizer_TTSD_V0_hf", trust_remote_code=True)
34
+ model = AutoModel.from_pretrained("OpenMOSS-Team/MOSS-TTSD-v0.7", trust_remote_code=True, device_map="auto").eval()
35
 
36
  data = [{
37
  "base_path": "/path/to/audio/files",
modeling_moss_ttsd.py CHANGED
@@ -203,7 +203,7 @@ class MossTTSDGenerationMixin(GenerationMixin):
203
  stopping_criteria: StoppingCriteriaList,
204
  generation_config: GenerationConfig,
205
  synced_gpus: bool,
206
- streamer: Optional[BaseStreamer],
207
  **model_kwargs,
208
  ) -> Union[GenerateDecoderOnlyOutput, torch.LongTensor]:
209
  """Sample method for multi-channel TTS generation."""
 
203
  stopping_criteria: StoppingCriteriaList,
204
  generation_config: GenerationConfig,
205
  synced_gpus: bool,
206
+ streamer: Optional[BaseStreamer]=None,
207
  **model_kwargs,
208
  ) -> Union[GenerateDecoderOnlyOutput, torch.LongTensor]:
209
  """Sample method for multi-channel TTS generation."""