d3rp3tt3 / AI-Trivia-ChatBot-Project-3

3 stars 1 forks source link

Add initial model #3

Closed d3rp3tt3 closed 1 week ago

d3rp3tt3 commented 1 week ago

Overview

This change adds the initial LLM model.

Details

Modules

from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain.embeddings import HuggingFaceEmbeddings
from langchain.vectorstores import Chroma
from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
from langchain import HuggingFacePipeline
from langchain.chains.question_answering import load_qa_chain
from langchain.prompts import PromptTemplate