Spaces:
Sleeping
Sleeping
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
|