dinhanhx/gpu-friendly-rag

A very GPU-friendly implementation

0

stars

2

commits

Python

primary language

Oct 29, 2024

updated

nlp
python
retrieval-augmented-generation

README

A very GPU-friendly RAG implementation

A simple RAG tool for asking things related to a pdf file (in this project, it's about Don Quixote)

This project demonstrates how things can be implemented with GPU-friendly database and models. It uses

I use a NVIDIA A30 GPU to run this.

Setup

Python 3.10, miniconda

Install required packages, please see setup.sh for extra information

bash setup.sh

Run

It is strongly advised to reach each .py file before running any command. By doing so, you get to understand the project more.

At root project, to setup the vector database

python setup_db.py

At root project, to query something then have a language model answer

python rag.py

Great researcher/developer-friendly RAG frameworks

Contributors

dinhanhx

2 commits

dinhanhx/gpu-friendly-rag

A very GPU-friendly implementation

0

stars

2

commits

Python

primary language

Oct 29, 2024

updated

nlp
python
retrieval-augmented-generation

README

A very GPU-friendly RAG implementation

A simple RAG tool for asking things related to a pdf file (in this project, it's about Don Quixote)

This project demonstrates how things can be implemented with GPU-friendly database and models. It uses

I use a NVIDIA A30 GPU to run this.

Setup

Python 3.10, miniconda

Install required packages, please see setup.sh for extra information

bash setup.sh

Run

It is strongly advised to reach each .py file before running any command. By doing so, you get to understand the project more.

At root project, to setup the vector database

python setup_db.py

At root project, to query something then have a language model answer

python rag.py

Great researcher/developer-friendly RAG frameworks

Contributors

dinhanhx

2 commits

Languages

Python

91.9%

Shell

8.1%