Add exclusive music GPU mode beside LLM and video

2026-09-30 22:13:31 +02:00
parent 0ca40bc624
commit 3aa2301a53
12 changed files with 46 additions and 16 deletions
@@ -67,6 +67,9 @@ class EndpointTests(unittest.TestCase):
  self.enable('music')
  self.assertEqual(self.request('/v1/audio/music/models')[1]['data'][0]['id'],'Music')
  self.assertNotIn('Music',[x['id'] for x in self.request()[1]['data']])
+  self.assertEqual(self.request('/v1/audio/music',dict(model='Music',lyrics='Synthetic',style='pop'))[0],503)
+  self.ep.scheduler.gpu_mode='music'
+  self.assertEqual(self.request('/v1/chat/completions',dict(model='alpha',messages=[dict(role='user',content='Synthetic')]))[0],503)
  conn=http.client.HTTPConnection('127.0.0.1',self.port,timeout=5)
  conn.request('POST','/v1/audio/music',json.dumps(dict(model='Music',lyrics='Synthetic',style='pop')),{'Content-Type':'application/json','Authorization':'Bearer '+'A'*32})
  response=conn.getresponse();self.assertEqual(response.status,200);self.assertEqual(response.getheader('Content-Type'),'audio/wav');self.assertEqual(response.read(),b'RIFFsynthetic-WAV');conn.close()