Colab

[christophschuhmann]

I tried to make a colab notebook, but it gives this error, even after installing all requirements from the readme:
https://colab.research.google.com/drive/1h5jOxlNPmbQIivY_wqu0481F0_x6Gw19?usp=sharing

Maybe you can get it to work. :)

import os
os.chdir("/content/VoiceStar")

!python inference_commandline.py
--reference_speech "./demo/5895_34622_000026_000002.wav"
--target_text "I cannot believe that the same model can also do text to speech synthesis too! And you know what? this audio is 8 seconds long."
--target_duration 8

Traceback (most recent call last):
File "/content/VoiceStar/inference_commandline.py", line 15, in
from models import voice_star
File "/content/VoiceStar/models/voice_star.py", line 9, in
from torchmetrics.classification import MulticlassAccuracy
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/init.py", line 37, in
from torchmetrics import functional # noqa: E402
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/functional/init.py", line 56, in
from torchmetrics.functional.image._deprecated import (
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/functional/image/init.py", line 14, in
from torchmetrics.functional.image.arniqa import arniqa
File "/usr/local/lib/python3.11/dist-packages/torchmetrics/functional/image/arniqa.py", line 31, in
from torchvision import transforms
File "/usr/local/lib/python3.11/dist-packages/torchvision/init.py", line 10, in
from torchvision import _meta_registrations, datasets, io, models, ops, transforms, utils # usort:skip
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torchvision/_meta_registrations.py", line 163, in
@torch.library.register_fake("torchvision::nms")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torch/library.py", line 795, in register
use_lib._register_fake(op_name, func, _stacklevel=stacklevel + 1)
File "/usr/local/lib/python3.11/dist-packages/torch/library.py", line 184, in _register_fake
handle = entry.fake_impl.register(func_to_register, source)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/usr/local/lib/python3.11/dist-packages/torch/_library/fake_impl.py", line 31, in register
if torch._C._dispatch_has_kernel_for_dispatch_key(self.qualname, "Meta"):
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
RuntimeError: operator torchvision::nms does not exist

[RafaelGodoyEbert]

!pip uninstall -y torch torchaudio torchvision
!pip install torch==2.5.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu124
!pip install torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124

or https://colab.research.google.com/drive/1tQkJsAenGgsDUSLeLxjA4AKewAGaLJzk?usp=sharing