OpenMOSS-Team
/

MOSS-VoiceGenerator

Model card Files Files and versions

CaasiHUANG commited on 3 days ago

Commit

84d8b58

·

verified ·

1 Parent(s): db8a984

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -78,7 +78,7 @@ torch.backends.cuda.enable_flash_sdp(True)
 torch.backends.cuda.enable_mem_efficient_sdp(True)
 torch.backends.cuda.enable_math_sdp(True)
-pretrained_model_name_or_path = "OpenMOSS-Team/MOSS-Voice-Generator"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16 if device == "cuda" else torch.float32

 torch.backends.cuda.enable_mem_efficient_sdp(True)
 torch.backends.cuda.enable_math_sdp(True)
+pretrained_model_name_or_path = "OpenMOSS-Team/MOSS-VoiceGenerator"
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.bfloat16 if device == "cuda" else torch.float32