Spaces:

OuroborosM
/

STLA-BABY

Runtime error

OuroborosM commited on Aug 15, 2023

Commit

3f28a81

1 Parent(s): af5e898

update embedding

Files changed (1) hide show

app.py CHANGED Viewed

@@ -527,7 +527,6 @@ device = 'cpu'
 embed_model = HuggingFaceEmbeddings(
     model_name=embed_model_id,
     model_kwargs={'device': device},
-    encode_kwargs={'device': device, 'batch_size': 32}
 )
 embeddings = embeddings_openai

 embed_model = HuggingFaceEmbeddings(
     model_name=embed_model_id,
     model_kwargs={'device': device},
 )
 embeddings = embeddings_openai