Avoid YuEGP pinned-memory exhaustion on 32GB Windows hosts

2026-09-08 07:04:13 -05:00
parent 7665285a0d
commit 4c97980646
3 changed files with 35 additions and 2 deletions
@@ -11,6 +11,16 @@ spec.loader.exec_module(worker)


 class WorkerTests(unittest.TestCase):
+    def test_windows_memory_fix_preserves_profile_and_precision(self):
+        options = worker.memory_profile_options(1, False, 'win32', 32 * 1024 ** 3)
+        self.assertFalse(options['pinnedMemory'])
+        self.assertEqual(options['profile_no'], 1)
+        self.assertFalse(options['quantizeTransformer'])
+        self.assertFalse(options['compile'])
+        self.assertNotIn('budgets', options)
+        self.assertNotIn('pinnedMemory', worker.memory_profile_options(1, False, 'linux', 32 * 1024 ** 3))
+        self.assertTrue(worker.memory_profile_options(3, False, 'win32', 32 * 1024 ** 3)['quantizeTransformer'])
+
    def test_lyrics_preserve_words_and_normalize_ui_headings(self):
        self.assertEqual(worker.normalize_lyrics('[Pre-Chorus]\nEvery word stays\n[Outro]\n'), '[prechorus]\nEvery word stays\n\n')
        self.assertEqual(worker.normalize_lyrics('[Verse 1]\nHello'), '[verse1]\nHello\n\n')