Enable JAX compilation for streaming tasks
All checks were successful
Build and Deploy / build-and-deploy (push) Successful in 2m50s
All checks were successful
Build and Deploy / build-and-deploy (push) Successful in 2m50s
The StreamInterpreter was created without use_jax=True, so the JAX compiler was never activated for production rendering. Desktop testing had this enabled but the celery task path did not. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -354,7 +354,7 @@ def run_stream(
|
||||
checkpoint = None
|
||||
|
||||
# Create interpreter (pass actor_id for friendly name resolution)
|
||||
interp = StreamInterpreter(str(recipe_path), actor_id=actor_id)
|
||||
interp = StreamInterpreter(str(recipe_path), actor_id=actor_id, use_jax=True)
|
||||
|
||||
# Set primitive library directory explicitly
|
||||
interp.primitive_lib_dir = sexp_effects_dir / "primitive_libs"
|
||||
|
||||
Reference in New Issue
Block a user