diff --git a/utils.py b/utils.py index 592029b4..d93d839b 100644 --- a/utils.py +++ b/utils.py @@ -49,7 +49,7 @@ GROQ_LLM = ChatGroq(temperature=0, model_name="llama3-8b-8192", max_tokens=100) # loading the embedding model def load_embedding_model(): model_name = "BAAI/bge-small-en" - model_kwargs = {"device": "cuda"} #can also be cpu + model_kwargs = {"device": "cpu"} #can also be cpu encode_kwargs = {"normalize_embeddings": True} embeddings = HuggingFaceBgeEmbeddings( model_name=model_name, model_kwargs=model_kwargs, encode_kwargs=encode_kwargs