We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c1980f7 commit e91039dCopy full SHA for e91039d
includes/dependencies.jinja
@@ -71,7 +71,7 @@
71
{%- if 'timm' in transformers_deps %}
72
"timm>=1.0.15",
73
{%- endif %}
74
- {%- if 'pytorch_cpu' in dl_pytorch_deps or 'pytorch_cu118' in dl_pytorch_deps or 'pytorch_cu121' in dl_pytorch_deps or 'pytorch_cu124' in dl_pytorch_deps %}
+ {%- if 'pytorch_cpu' in dl_pytorch_deps or 'pytorch_cu129' in dl_pytorch_deps or 'pytorch_cu128' in dl_pytorch_deps or 'pytorch_cu126' in dl_pytorch_deps or 'pytorch_cu124' in dl_pytorch_deps or 'pytorch_cu121' in dl_pytorch_deps or 'pytorch_cu118' in dl_pytorch_deps %}
75
"torch>=2.3.0",
76
77
{%- if 'torchvision' in dl_pytorch_deps %}
0 commit comments