Torchvision dependency issue

[adamkarvonen]

Most times that I install nnsight on a new machine, I get this error:

Traceback (most recent call last):
File "/workspace/dictionary_learning/demo.py", line 2, in
from nnsight import LanguageModel
File "/opt/conda/lib/python3.10/site-packages/nnsight/init.py", line 35, in
from .models.LanguageModel import LanguageModel
File "/opt/conda/lib/python3.10/site-packages/nnsight/models/LanguageModel.py", line 8, in
from transformers import (
File "", line 1075, in _handle_fromlist
File "/opt/conda/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1781, in getattr
module = self._get_module(self._class_to_module[name])
File "/opt/conda/lib/python3.10/site-packages/transformers/utils/import_utils.py", line 1795, in _get_module
raise RuntimeError(
RuntimeError: Failed to import transformers.modeling_utils because of the following error (look up to see its traceback):

Because the default version of torchvision installed is 0.17. If I install torchvision 0.20.1, the error goes away.

The most recent time this happened was with torch version 2.5.1 and nnsight version 0.3.7.