Coordinate shared GPU access across studio instances

server/utils/imageComfyLifecycle.ts

@@ -1,3 +1,4 @@
+import { assertSharedGpu, sharedGpuHeaders, withSharedGpuStart } from '~/server/utils/sharedGpu'
 import { getImageComfyHost, setImageComfyHostOverride, imageComfyConfigured, getBeastImageHost, getSidecarImageHost, withImageComfyHost, type ImageEditBox } from '~/server/utils/imageComfy'
 
 export type ImageComfyState = 'online' | 'booting' | 'busy' | 'starting' | 'offline'
@@ -92,12 +93,14 @@ export async function fetchImageQueue() {
 async function controlRequest(path: string, method = 'GET', timeoutMs = 5000) {
   const { controlUrl, controlToken } = settings()
   if (!controlUrl) return null
+  if (method !== 'GET') await assertSharedGpu()
   try {
     const res = await fetch(`${controlUrl}${path}`, {
       method,
       headers: {
         Accept: 'application/json',
-        ...(controlToken ? { Authorization: `Bearer ${controlToken}` } : {})
+        ...(controlToken ? { Authorization: `Bearer ${controlToken}` } : {}),
+        ...(method !== 'GET' ? sharedGpuHeaders() : {})
       },
       signal: AbortSignal.timeout(timeoutMs)
     })
@@ -238,7 +241,7 @@ async function ensureUnlocked(onStatus: StatusFn) {
 }
 
 export function ensureImageComfyReady(onStatus: StatusFn = () => undefined) {
-  const run = gate.then(() => ensureUnlocked(onStatus))
+  const run = gate.then(() => withSharedGpuStart(() => ensureUnlocked(onStatus), async () => { await assertSharedGpu() }))
   gate = run.then(() => undefined, () => undefined)
   return run
 }