Upload 7 files
Browse files- embedding_utils.py +2 -2
embedding_utils.py
CHANGED
@@ -87,7 +87,7 @@ def parallel_generate_embeddings(
|
|
87 |
|
88 |
processed = 0
|
89 |
current_batch_size = batch_size
|
90 |
-
max_workers =
|
91 |
|
92 |
logger.info(f"Starting embedding generation for {total_docs} documents")
|
93 |
if callback:
|
@@ -122,7 +122,7 @@ def parallel_generate_embeddings(
|
|
122 |
logger.error(f"Error processing futures: {str(e)}")
|
123 |
# Reduce batch size and workers on error
|
124 |
current_batch_size = max(5, current_batch_size - 5)
|
125 |
-
max_workers = max(
|
126 |
logger.info(f"Reduced batch size to {current_batch_size}, workers to {max_workers}")
|
127 |
|
128 |
# Process remaining batch
|
|
|
87 |
|
88 |
processed = 0
|
89 |
current_batch_size = batch_size
|
90 |
+
max_workers = 10 # Start with fewer workers
|
91 |
|
92 |
logger.info(f"Starting embedding generation for {total_docs} documents")
|
93 |
if callback:
|
|
|
122 |
logger.error(f"Error processing futures: {str(e)}")
|
123 |
# Reduce batch size and workers on error
|
124 |
current_batch_size = max(5, current_batch_size - 5)
|
125 |
+
max_workers = max(5, max_workers - 2)
|
126 |
logger.info(f"Reduced batch size to {current_batch_size}, workers to {max_workers}")
|
127 |
|
128 |
# Process remaining batch
|