Add worker initialization and improve multiprocessing setup in audio_utils.py

This commit introduces a new `worker_init` function to ensure correct umask settings for worker processes in the audio generation workflow. Additionally, the multiprocessing start method is now set conditionally to avoid runtime errors if it has already been established. These changes enhance the robustness and reliability of the audio generation process, particularly in multi-process environments.

utils/audio_utils.py

@@ -7,6 +7,11 @@ import random
 import multiprocessing as mp
 import stat
 
+def worker_init():
+    """Initialize each worker process"""
+    # Ensure worker processes have correct umask
+    os.umask(0)
+
 def ensure_directory_permissions(directory):
     """Ensure the directory has the correct permissions"""
     try:
@@ -71,8 +76,11 @@ def process_audio_generations(descriptions, model_name, generate_fn, prepare_arg
         num_variations: Number of variations to generate per prompt
         num_processes: Number of parallel processes to use
     """
-    # Set start method for multiprocessing
-    mp.set_start_method('spawn', force=True)
+    # Set start method for multiprocessing if it hasn't been set
+    try:
+        mp.set_start_method('spawn')
+    except RuntimeError:
+        pass  # method already set
     
     if not descriptions:
         print('At least one prompt should be provided')
@@ -86,11 +94,6 @@ def process_audio_generations(descriptions, model_name, generate_fn, prepare_arg
     ensure_directory_permissions(base_output_dir)
     ensure_directory_permissions(model_dir)
     
-    def worker_init():
-        """Initialize each worker process"""
-        # Ensure worker processes have correct umask
-        os.umask(0)
-    
     for description in descriptions:
         # Create model-specific directory for each prompt
         prompt_dir = setup_generation_dir(model_dir, description)