Sfoglia il codice sorgente

examples : install torch-cpu for model conversion tool/example (#15475)

Signed-off-by: Jie Fu <jiefu@tencent.com>
Jie Fu (傅杰) 5 mesi fa
parent
commit
ad294df03f
1 ha cambiato i file con 1 aggiunte e 0 eliminazioni
  1. 1 0
      examples/model-conversion/requirements.txt

+ 1 - 0
examples/model-conversion/requirements.txt

@@ -1,3 +1,4 @@
+--extra-index-url https://download.pytorch.org/whl/cpu
 torch~=2.6.0
 torchvision~=0.21.0
 transformers~=4.55.0