art-dag/celery
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Re-enable GPU queue routing after image rebuild

Browse Source 
Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
This commit is contained in:
giles
2026-02-03 20:38:37 +00:00
parent 4930eb99ad
commit 09d5359725
1 changed files with 2 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
app/routers/runs.py
View File

@@ -227,11 +227,7 @@ async def create_stream_run(
logger.warning(f"Failed to store recipe in cache: {e}") logger.warning(f"Failed to store recipe in cache: {e}")
# Continue anyway - run will still work, just won't appear in /recipes # Continue anyway - run will still work, just won't appear in /recipes
# Submit Celery task - uses GPU queue if NVIDIA available, otherwise default # Submit Celery task to GPU queue for hardware-accelerated rendering
# Note: GPU worker needs image rebuild for IPFS HLS support, using default for now
import os
use_gpu_queue = os.environ.get("USE_GPU_QUEUE", "false").lower() == "true"
task_queue = 'gpu' if use_gpu_queue else None # None = default queue
task = run_stream.apply_async( task = run_stream.apply_async(
kwargs=dict( kwargs=dict(
run_id=run_id, run_id=run_id,
@@ -243,7 +239,7 @@ async def create_stream_run(
sources_sexp=request.sources_sexp, sources_sexp=request.sources_sexp,
audio_sexp=request.audio_sexp, audio_sexp=request.audio_sexp,
), ),
queue=task_queue, queue='gpu',
) )
# Store in database for durability # Store in database for durability