jaydeepthik / Gemma-RAG Public

Notifications You must be signed in to change notification settings
Fork 0
Star 1

Retrieval Augmented Generation (RAG) using FAISS and Gemma 2b instruction tuned model for answering your questions regarding movies.

1 star 0 forks Branches Tags Activity

Name		Name	Last commit message	Last commit date
Latest commit History 5 Commits
Gemma_rag_movies.ipynb		Gemma_rag_movies.ipynb
README.md		README.md

Repository files navigation

Gemma-RAG

Retrieval Augmented Generation (RAG) using FAISS and Gemma 2b instruction tuned model from HuggingFace for answering your questions regarding movies.

Details

Dataset: "MongoDB/embedded_movies"
Embedding: "thenlper/gte-small"
VectorDB: FAISS
Generator LLM: "google/gemma-2b-it"

Libraries

accelerate
bitsandbytes
transformers
sentence-transformers
faiss-gpu
datasets
langchain