Skip to content

timm library missing in requirements.txt #3

Open
@RJacobArthrex

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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions