diff --git a/pyproject.toml b/pyproject.toml index 06624560..c9384912 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,7 +23,7 @@ dependencies = [ 'pillow', 'pretrainedmodels==0.7.4', 'six', - 'timm==0.9.7', + 'timm==1.0.9', 'torch', 'torchvision>=0.5', 'tqdm', diff --git a/requirements.txt b/requirements.txt index aa700f60..e8fd0ce2 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,7 +1,7 @@ torchvision>=0.5.0 pretrainedmodels==0.7.4 efficientnet-pytorch==0.7.1 -timm==0.9.7 +timm==1.0.9 huggingface_hub>=0.24.6 tqdm