Spaces:

GIZ
/

audit_assistant

Running on T4

ppsingh commited on Feb 20

Commit

2f7f80c

verified ·

1 Parent(s): 40cbe96

Update auditqa/process_chunks.py

Files changed (1) hide show

auditqa/process_chunks.py CHANGED Viewed

@@ -61,7 +61,6 @@ def load_chunks():
     # define embedding model
     embeddings = HuggingFaceEmbeddings(
         model_kwargs = {'device': device},
-        show_progress = True,
         multi_process = True,
         encode_kwargs = {'normalize_embeddings': bool(int(config.get('retriever','NORMALIZE')))},
         model_name=config.get('retriever','MODEL')

     # define embedding model
     embeddings = HuggingFaceEmbeddings(
         model_kwargs = {'device': device},
         multi_process = True,
         encode_kwargs = {'normalize_embeddings': bool(int(config.get('retriever','NORMALIZE')))},
         model_name=config.get('retriever','MODEL')