From b609f0681bbdf7ab15d05b8bf90e6a58e0d8d5ed Mon Sep 17 00:00:00 2001
From: Joo-Won Jung <sanori@gmail.com>
Date: Tue, 2 Apr 2024 02:51:50 +0900
Subject: [PATCH] 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>
---
 Pipfile.torchidx | 36 ++++++++++++++++++++++++++++++++++++
 README.md        |  3 +++
 2 files changed, 39 insertions(+)
 create mode 100644 Pipfile.torchidx

diff --git a/Pipfile.torchidx b/Pipfile.torchidx
new file mode 100644
index 0000000000..ae01a473fb
--- /dev/null
+++ b/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"
diff --git a/README.md b/README.md
index 7eb6006774..029c40d62a 100644
--- a/README.md
+++ b/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