Skip to content

Commit e91039d

Browse files
committed
🐛 fix: add torch as a dep when choosing newer version of CUDA
1 parent c1980f7 commit e91039d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

includes/dependencies.jinja

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -71,7 +71,7 @@
7171
{%- if 'timm' in transformers_deps %}
7272
"timm>=1.0.15",
7373
{%- 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 %}
74+
{%- 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 %}
7575
"torch>=2.3.0",
7676
{%- endif %}
7777
{%- if 'torchvision' in dl_pytorch_deps %}

0 commit comments

Comments
 (0)