Open
Description
Issue found on:
Python Version: 3.10.14
OS: Linux
Description:
After installing latest (1.0.1) version via pip install timm-3d
, when I try to run import timm_3d
, following error occurs:
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
File "/anaconda/envs/azureml_py38_PT_TF/lib/python3.10/site-packages/timm_3d/__init__.py", line 3, in <module>
from .models import create_model, list_models, list_pretrained, is_model, list_modules, model_entrypoint, \
File "/anaconda/envs/azureml_py38_PT_TF/lib/python3.10/site-packages/timm_3d/models/__init__.py", line 8, in <module>
from .convnext import *
File "/anaconda/envs/azureml_py38_PT_TF/lib/python3.10/site-packages/timm_3d/models/convnext.py", line 51, in <module>
from ._builder import build_model_with_cfg
File "/anaconda/envs/azureml_py38_PT_TF/lib/python3.10/site-packages/timm_3d/models/_builder.py", line 13, in <module>
from timm_3d.models._hub import has_hf_hub, download_cached_file, check_cached_file, load_state_dict_from_hf
File "/anaconda/envs/azureml_py38_PT_TF/lib/python3.10/site-packages/timm_3d/models/_hub.py", line 29, in <module>
from timm import __version__
ModuleNotFoundError: No module named 'timm'
Reason for error:
This is due to requirements.txt
not containing timm
:
torch>=1.7
torchvision
pyyaml
huggingface_hub
safetensors>=0.2
Possible Solution
I know that I can fix this by just doing pip install timm
but is it possible for you to add timm
to requirements.txt? This way during pip install dependencies are installed. Example:
requirements.txt
torch>=1.7
torchvision
pyyaml
huggingface_hub
safetensors>=0.2
timm
Metadata
Assignees
Labels
No labels