airabbitX commited on
Commit
7301668
·
verified ·
1 Parent(s): bbccab6

Upload 7 files

Browse files
Files changed (1) hide show
  1. 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 = 5 # Start with fewer 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(3, max_workers - 2)
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