Add files via upload

gui_data/constants.py

@@ -502,6 +502,7 @@ MP3 = 'MP3'
 
 MP3_BIT_RATES = ('96k', '128k', '160k', '224k', '256k', '320k')
 WAV_TYPE = ('PCM_U8', 'PCM_16', 'PCM_24', 'PCM_32', '32-bit Float', '64-bit Float')
+CUDA_TYPE = (DEFAULT, '0', '1', '2', '3', '4', '5', '6', '7', '8')
 
 SELECT_SAVED_SET = 'Choose Option'
 SAVE_SETTINGS = 'Save Current Settings'
@@ -646,12 +647,13 @@ DEFAULT_DATA = {
         'semitone_shift': '0',#
         'save_format': WAV,
         'wav_type_set': 'PCM_16',
+        'cuda_set': DEFAULT,
         'user_code': '',
         'export_path': '',
         'input_paths': [],
         'lastDir': None,
         'time_window': "3",
-        'intro_analysis': "Default",
+        'intro_analysis': DEFAULT,
         'db_analysis': "Medium",
         'fileOneEntry': '',
         'fileOneEntry_Full': '',
@@ -756,6 +758,7 @@ SETTING_CHECK = ('vr_model',
                'semitone_shift',#
                'save_format',
                'wav_type_set',
+               'cuda_set',
                'user_code',
                'is_gpu_conversion',
                'is_normalization',
@@ -1018,6 +1021,7 @@ else:
 IS_TIME_CORRECTION_HELP = ('When checked, the output will retain the original BPM of the input.')
 SAVE_STEM_ONLY_HELP = 'Allows the user to save only the selected stem.'
 IS_NORMALIZATION_HELP = 'Normalizes output to prevent clipping.'
+IS_CUDA_SELECT_HELP = "If you have more than one GPU, you can pick which one to use for processing."
 CROP_SIZE_HELP = '**Only compatible with select models only!**\n\n Setting should match training crop-size value. Leave as is if unsure.'
 IS_TTA_HELP = ('This option performs Test-Time-Augmentation to improve the separation quality.\n\n'
                'Note: Having this selected will increase the time it takes to complete a conversion')
@@ -1502,6 +1506,7 @@ VOLUME_COMPENSATION_TEXT = 'Volume Compensation'
 VR_51_MODEL_TEXT = 'VR 5.1 Model'
 VR_ARCH_TEXT = 'VR Arch'
 WAV_TYPE_TEXT = 'Wav Type'
+CUDA_NUM_TEXT = 'GPU Device'
 WINDOW_SIZE_TEXT = 'Window Size'
 YES_TEXT = 'Yes'
 VERIFY_INPUTS_TEXT = 'Verify Inputs'

gui_data/error_handling.py

@@ -18,6 +18,7 @@ SYSTEM_MEMORY_ERROR = "DefaultCPUAllocator: not enough memory"
 MISSING_MODEL_ERROR = "'NoneType\' object has no attribute \'model_basename\'"
 ARRAY_SIZE_ERROR = "ValueError: \"array is too big; `arr.size * arr.dtype.itemsize` is larger than the maximum possible size.\""
 GPU_INCOMPATIBLE_ERROR = "no kernel image is available for execution on the device"
+SELECT_CORRECT_GPU = "CUDA kernel errors might be asynchronously reported at some other API call,so the stacktrace below might be incorrect."
 
 CONTACT_DEV = 'If this error persists, please contact the developers with the error details.'
 
@@ -27,7 +28,7 @@ ERROR_MAPPER = {
                         'Please close any GPU intensive applications and try again.\n' + 
                         'If the error persists, your GPU might not be supported.') ,
     CUDA_RUNTIME_ERROR:
-                        (f'Your PC cannot process this audio file with the chunk size selected. Please lower the chunk size and try again.\n\n{CONTACT_DEV}'),
+                        (f'Your PC cannot process this audio file with the segment size selected. Please lower the segment size and try again.\n\n{CONTACT_DEV}'),
     DEMUCS_MODEL_MISSING_ERROR:
                         ('The selected Demucs model is missing. ' + 
                         'Please download the model or make sure it is in the correct directory.'),
@@ -43,7 +44,7 @@ ERROR_MAPPER = {
                         (f'Missing file error raised. Please address the error and try again.\n\n{CONTACT_DEV}'),
     MDX_MEMORY_ERROR:
                         ('The application was unable to allocate enough GPU memory to use this model.\n\n' + 
-                        'Please do the following:\n\n1. Close any GPU intensive applications.\n2. Lower the set chunk size.\n3. Then try again.\n\n' + 
+                        'Please do the following:\n\n1. Close any GPU intensive applications.\n2. Lower the set segment size.\n3. Then try again.\n\n' + 
                         'If the error persists, your GPU might not be supported.'),
     MDX_MODEL_MISSING:
                         ('The application could not detect this MDX-Net model on your system. ' + 
@@ -51,7 +52,7 @@ ERROR_MAPPER = {
                         'If the error persists, please reinstall application or contact the developers.'),
     MDX_RUNTIME_ERROR:
                         ('The application was unable to allocate enough GPU memory to use this model.\n\n' + 
-                        'Please do the following:\n\n1. Close any GPU intensive applications.\n2. Lower the set chunk size.\n3. Then try again.\n\n' + 
+                        'Please do the following:\n\n1. Close any GPU intensive applications.\n2. Lower the set segment size.\n3. Then try again.\n\n' + 
                         'If the error persists, your GPU might not be supported.'),
     WINDOW_SIZE_ERROR:
                         ('Invalid window size.\n\n' + 
@@ -70,6 +71,9 @@ ERROR_MAPPER = {
     GPU_INCOMPATIBLE_ERROR:
                         ('This process is not compatible with your GPU.\n\n' + 
                         'Please uncheck \"GPU Conversion\" and try again'),
+    SELECT_CORRECT_GPU:
+                        ('Make sure you\'ve chosen the correct GPU.\n\n'
+                        'Go to the "Settings Guide", click the "Additional Settings" tab and select the correct GPU device.'),
     ARRAY_SIZE_ERROR:
                         ('The application was not able to process the given audiofile. Please convert the audiofile to another format and try again.'),
 }