Skip to content
/ Gemma-RAG Public

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

3 stars 0 forks Branches Tags Activity
Star
Notifications You must be signed in to change notification settings

jaydeepthik/Gemma-RAG

BranchesTags

Folders and files

NameName
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

About

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

Topics

gemma faiss rag retrieval-augmented-generation

Resources

Readme
Activity

Stars

3 stars

Watchers

1 watching

Forks

0 forks
Report repository

Releases

No releases published

Packages

No packages published

Languages