Ver código fonte

Add an Pipfile example for AMD GPU or CPU only users (#510)

* refactor: add Pipfile example to select various PyTorch package

* Update README.md

---------

Co-authored-by: Robert Brennan <accounts@rbren.io>
Joo-Won Jung 1 ano atrás
pai
commit
b609f0681b
2 arquivos alterados com 39 adições e 0 exclusões
  1. 36 0
      Pipfile.torchidx
  2. 3 0
      README.md

+ 36 - 0
Pipfile.torchidx

@@ -0,0 +1,36 @@
+[[source]]
+url = "https://pypi.org/simple"
+verify_ssl = true
+name = "pypi"
+
+[[source]]
+url = "https://download.pytorch.org/whl/cpu"
+verify_ssl = true
+name = "pytorch"
+
+[packages]
+torch = {version = "*", index = "pytorch"}
+datasets = "*"
+pandas = "*"
+litellm = "*"
+termcolor = "*"
+seaborn = "*"
+docker = "*"
+fastapi = "*"
+uvicorn = {extras = ["standard"], version = "*"}
+ruff = "*"
+mypy = "*"
+llama-index = "*"
+llama-index-vector-stores-chroma = "*"
+chromadb = "*"
+llama-index-embeddings-huggingface = "*"
+llama-index-embeddings-azure-openai = "*"
+llama-index-embeddings-ollama = "*"
+google-generativeai = "*"
+toml = "*"
+json_repair = "*"
+
+[dev-packages]
+
+[requires]
+python_version = "3.11"

+ 3 - 0
README.md

@@ -47,6 +47,9 @@ If `pipenv` doesn't work for you, you can also run:
 python -m pipenv requirements > requirements.txt && python -m pip install -r requirements.txt
 ```
 
+If you're seeing installation errors due to `torch`, try using `Pipfile.torchidx` instead of `Pipfile`.
+Please consult the value of `--index-url` in [PyTorch install command browser](https://pytorch.org/get-started/locally/#start-locally) to find appropriate package index URL for you.
+
 Then, in a second terminal, start the frontend:
 ```bash
 cd frontend