Skip to content

Commit dcaed95

Browse files
authored
remove invalid references to facebook/parler-tts-small (#132)
1 parent 8e465f1 commit dcaed95

File tree

2 files changed

+11
-11
lines changed

2 files changed

+11
-11
lines changed

parler_tts/configuration_parler_tts.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020

2121
logger = logging.get_logger(__name__)
2222

23-
MUSICGEN_PRETRAINED_CONFIG_ARCHIVE_MAP = {
24-
"facebook/parler_tts-small": "https://huggingface.co/facebook/parler_tts-small/resolve/main/config.json",
23+
PARLER_TTS_PRETRAINED_CONFIG_ARCHIVE_MAP = {
24+
"parler-tts/parler-tts-mini-v1": "https://huggingface.co/parler-tts/parler-tts-mini-v1/resolve/main/config.json",
2525
# See all ParlerTTS models at https://huggingface.co/models?filter=parler_tts
2626
}
2727

@@ -31,7 +31,7 @@ class ParlerTTSDecoderConfig(PretrainedConfig):
3131
This is the configuration class to store the configuration of an [`ParlerTTSDecoder`]. It is used to instantiate a
3232
Parler-TTS decoder according to the specified arguments, defining the model architecture. Instantiating a
3333
configuration with the defaults will yield a similar configuration to that of the Parler-TTS
34-
[facebook/parler_tts-small](https://huggingface.co/facebook/parler_tts-small) architecture.
34+
[parler-tts/parler-tts-mini-v1](https://huggingface.co/parler-tts/parler-tts-mini-v1) architecture.
3535
3636
Configuration objects inherit from [`PretrainedConfig`] and can be used to control the model outputs. Read the
3737
documentation from [`PretrainedConfig`] for more information.
@@ -203,7 +203,7 @@ class ParlerTTSConfig(PretrainedConfig):
203203
... text_encoder_config, audio_encoder_config, decoder_config
204204
... )
205205
206-
>>> # Initializing a ParlerTTSForConditionalGeneration (with random weights) from the facebook/parler_tts-small style configuration
206+
>>> # Initializing a ParlerTTSForConditionalGeneration (with random weights) from the parler-tts/parler-tts-mini-v1 style configuration
207207
>>> model = ParlerTTSForConditionalGeneration(configuration)
208208
209209
>>> # Accessing the model configuration

parler_tts/modeling_parler_tts.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -77,10 +77,10 @@
7777
logger.warn("Flash attention 2 is not installed")
7878

7979
_CONFIG_FOR_DOC = "ParlerTTSConfig"
80-
_CHECKPOINT_FOR_DOC = "facebook/parler_tts-small"
80+
_CHECKPOINT_FOR_DOC = "parler-tts/parler-tts-mini-v1"
8181

8282
MUSICGEN_PRETRAINED_MODEL_ARCHIVE_LIST = [
83-
"facebook/parler_tts-small",
83+
"parler-tts/parler-tts-mini-v1",
8484
# See all ParlerTTS models at https://huggingface.co/models?filter=parler_tts
8585
]
8686

@@ -2357,7 +2357,7 @@ def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
23572357
```python
23582358
>>> from parler_tts import ParlerTTSForConditionalGeneration
23592359
2360-
>>> model = ParlerTTSForConditionalGeneration.from_pretrained("facebook/parler_tts-small")
2360+
>>> model = ParlerTTSForConditionalGeneration.from_pretrained("parler-tts/parler-tts-mini-v1")
23612361
```"""
23622362

23632363
# At the moment fast initialization is not supported for composite models
@@ -2411,7 +2411,7 @@ def from_sub_models_pretrained(
24112411
24122412
- A string, the *model id* of a pretrained model hosted inside a model repo on huggingface.co.
24132413
Valid model ids can be located at the root-level, like `gpt2`, or namespaced under a user or
2414-
organization name, like `facebook/parler_tts-small`.
2414+
organization name, like `parler-tts/parler-tts-mini-v1`.
24152415
- A path to a *directory* containing model weights saved using
24162416
[`~PreTrainedModel.save_pretrained`], e.g., `./my_model_directory/`.
24172417
@@ -2440,7 +2440,7 @@ def from_sub_models_pretrained(
24402440
>>> model = ParlerTTSForConditionalGeneration.from_sub_models_pretrained(
24412441
... text_encoder_pretrained_model_name_or_path="t5-base",
24422442
... audio_encoder_pretrained_model_name_or_path="facebook/encodec_24khz",
2443-
... decoder_pretrained_model_name_or_path="facebook/parler_tts-small",
2443+
... decoder_pretrained_model_name_or_path="parler-tts/parler-tts-mini-v1",
24442444
... )
24452445
>>> # saving model after fine-tuning
24462446
>>> model.save_pretrained("./parler_tts-ft")
@@ -2607,8 +2607,8 @@ def forward(
26072607
>>> from transformers import AutoProcessor, ParlerTTSForConditionalGeneration
26082608
>>> import torch
26092609
2610-
>>> processor = AutoProcessor.from_pretrained("facebook/parler_tts-small")
2611-
>>> model = ParlerTTSForConditionalGeneration.from_pretrained("facebook/parler_tts-small")
2610+
>>> processor = AutoProcessor.from_pretrained("parler-tts/parler-tts-mini-v1")
2611+
>>> model = ParlerTTSForConditionalGeneration.from_pretrained("parler-tts/parler-tts-mini-v1")
26122612
26132613
>>> inputs = processor(
26142614
... text=["80s pop track with bassy drums and synth", "90s rock song with loud guitars and heavy drums"],

0 commit comments

Comments
 (0)