Camb-ai / MARS5-TTS

MARS5 speech model (TTS) from CAMB.AI
https://www.camb.ai
GNU Affero General Public License v3.0
2.53k stars 206 forks source link

Google Colab failing to load model #48

Closed Veeeetzzzz closed 4 months ago

Veeeetzzzz commented 4 months ago

When I run the Google Colab notebook, it throws an error when I reach this command, I believe it's down to a corrupt safetensors installation

mars5, config_class = torch.hub.load('Camb-ai/mars5-tts', 'mars5_english', trust_repo=True)

image

I fixed it by manually upgrading safetensors and using this command

!pip install --upgrade safetensors

image

I'm planning to run this locally - but thought it might be useful to publish my fix and perhaps the Colab file can be updated

RF5 commented 4 months ago

Thanks for the catch, updated the colab notebook with new safetensors dependency.