bhardwaj08sarthak commited on
Commit
4157bc3
·
verified ·
1 Parent(s): 10c17b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -95,7 +95,7 @@ LOCAL_BASE.mkdir(parents=True, exist_ok=True)
95
  # Recreate the SAME embedding model used to build the index
96
  emb = HuggingFaceEmbeddings(
97
  model_name="google/embeddinggemma-300m",
98
- model_kwargs={"device": "cuda", "attn_implementation": "eager"},
99
  encode_kwargs={"normalize_embeddings": True},
100
  )
101
 
 
95
  # Recreate the SAME embedding model used to build the index
96
  emb = HuggingFaceEmbeddings(
97
  model_name="google/embeddinggemma-300m",
98
+ model_kwargs={"device": "cuda"},
99
  encode_kwargs={"normalize_embeddings": True},
100
  )
101