update_readme (#1)
Browse files- update readme (dc16cdd883bba3fbef64c793b0b5c04bc1f6f681)
Co-authored-by: 甯告垚 <[email protected]>
- README.md +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("
|
| 34 |
-
model = AutoModel.from_pretrained("
|
| 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."""
|