Add exclusive music GPU mode beside LLM and video
This commit is contained in:
+1
-1
@@ -15,7 +15,7 @@ class MusicTests(unittest.TestCase):
|
||||
profiles=Mock();profiles.status.return_value=dict(profiles=[p]);profiles._component.side_effect=lambda model,role,id:entries[id]
|
||||
cli=root/'cli';cli.write_text('#!/usr/bin/env python3\nimport sys,wave\np=sys.argv[sys.argv.index("--out")+1]\nwith wave.open(p,"wb") as w:w.setparams((1,2,8000,0,"NONE","none"));w.writeframes(bytes(1600))\n');cli.chmod(0o755)
|
||||
runtime=Mock();runtime.status.return_value=dict(installed=True);runtime.paths.return_value=(cli,root)
|
||||
scheduler=Mock();release=Mock();scheduler.image_reservation.return_value=release
|
||||
scheduler=Mock();scheduler.gpu_mode='music';release=Mock();scheduler.image_reservation.return_value=release
|
||||
return Music(root/'jobs',profiles,catalog,runtime,scheduler),release,p
|
||||
def wait(self,m):
|
||||
for _ in range(80):
|
||||
|
||||
Reference in New Issue
Block a user