version of Torch

#24
by xxbadarxx - opened

image.png

image.png

I met these errors when running with the pipeline API .Which version of Torch is required to run Gemma2?

Google org

Hi @xxbadarxx ,

Could you please upgrade the Torch library by running the following command:

        pip install --upgrade torch torchvision torchaudio

Additionally, ensure that your Torch version is 2.3.0 or higher, as the latest Transformers rely on cache_utils.py, which includes torch._dynamo

Thank you.

Sign up or log in to comment