소스 검색

py : use cpu-only torch in requirements.txt (#8335)

compilade 1 년 전
부모
커밋
d39130a398
3개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      examples/llava/requirements.txt
  2. 1 0
      requirements/requirements-convert_hf_to_gguf.txt
  3. 1 0
      requirements/requirements-convert_hf_to_gguf_update.txt

+ 1 - 0
examples/llava/requirements.txt

@@ -1,3 +1,4 @@
 -r ../../requirements/requirements-convert_legacy_llama.txt
+--extra-index-url https://download.pytorch.org/whl/cpu
 pillow~=10.2.0
 torch~=2.2.1

+ 1 - 0
requirements/requirements-convert_hf_to_gguf.txt

@@ -1,2 +1,3 @@
 -r ./requirements-convert_legacy_llama.txt
+--extra-index-url https://download.pytorch.org/whl/cpu
 torch~=2.2.1

+ 1 - 0
requirements/requirements-convert_hf_to_gguf_update.txt

@@ -1,2 +1,3 @@
 -r ./requirements-convert_legacy_llama.txt
+--extra-index-url https://download.pytorch.org/whl/cpu
 torch~=2.2.1