update readme
Browse files
README.md
CHANGED
|
@@ -218,9 +218,7 @@ def initial_config(tokenizer, model_name_or_path):
|
|
| 218 |
return generation_config
|
| 219 |
|
| 220 |
|
| 221 |
-
|
| 222 |
-
|
| 223 |
-
pretrained_model_name_or_path = "OpenMOSS-Team/MOSS-TTS"
|
| 224 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 225 |
dtype = torch.bfloat16 if device == "cuda" else torch.float32
|
| 226 |
|
|
@@ -305,8 +303,6 @@ conversations = [
|
|
| 305 |
],
|
| 306 |
]
|
| 307 |
|
| 308 |
-
|
| 309 |
-
|
| 310 |
model = AutoModel.from_pretrained(
|
| 311 |
pretrained_model_name_or_path,
|
| 312 |
trust_remote_code=True,
|
|
@@ -398,7 +394,7 @@ def initial_config(tokenizer, model_name_or_path):
|
|
| 398 |
return generation_config
|
| 399 |
|
| 400 |
|
| 401 |
-
pretrained_model_name_or_path = "OpenMOSS-Team/MOSS-TTS"
|
| 402 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 403 |
dtype = torch.bfloat16 if device == "cuda" else torch.float32
|
| 404 |
|
|
@@ -482,7 +478,6 @@ generation_config.layers = [
|
|
| 482 |
}
|
| 483 |
] * (model.channels - 1)
|
| 484 |
|
| 485 |
-
|
| 486 |
batch_size = 1
|
| 487 |
|
| 488 |
save_dir = Path("inference_root_moss_tts_local_transformer_continuation")
|
|
|
|
| 218 |
return generation_config
|
| 219 |
|
| 220 |
|
| 221 |
+
pretrained_model_name_or_path = "OpenMOSS-Team/MOSS-TTS-Local-Transformer"
|
|
|
|
|
|
|
| 222 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 223 |
dtype = torch.bfloat16 if device == "cuda" else torch.float32
|
| 224 |
|
|
|
|
| 303 |
],
|
| 304 |
]
|
| 305 |
|
|
|
|
|
|
|
| 306 |
model = AutoModel.from_pretrained(
|
| 307 |
pretrained_model_name_or_path,
|
| 308 |
trust_remote_code=True,
|
|
|
|
| 394 |
return generation_config
|
| 395 |
|
| 396 |
|
| 397 |
+
pretrained_model_name_or_path = "OpenMOSS-Team/MOSS-TTS-Local-Transformer"
|
| 398 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 399 |
dtype = torch.bfloat16 if device == "cuda" else torch.float32
|
| 400 |
|
|
|
|
| 478 |
}
|
| 479 |
] * (model.channels - 1)
|
| 480 |
|
|
|
|
| 481 |
batch_size = 1
|
| 482 |
|
| 483 |
save_dir = Path("inference_root_moss_tts_local_transformer_continuation")
|