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>
This commit is contained in:
Joo-Won Jung 2024-04-02 02:51:50 +09:00 committed by GitHub
parent 511afa12fe
commit b609f0681b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 39 additions and 0 deletions

36
Pipfile.torchidx Normal file
View File

@ -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"

View File

@ -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