magi_compile( dynamic_arg_dims={ "hidden_states": 0, "kv_buf": 2,

magi_compile( dynamic_arg_dims={"hidden_states": 0}, config_patch=_magi_config_patch,

magi_register_custom_op( "lightx2v::fp8_scaled_mm", infer_output_meta_fn=_fp8_scaled_mm_meta,

magi_register_custom_op( "lightx2v::qwen_rope_flashinfer", infer_output_meta_fn=_qwen_rope_met

magi_register_custom_op( "lightx2v::qwen_rope_torch", infer_output_meta_fn=_qwen_rope_meta,

magi_register_custom_op( "lightx2v::rms_norm", infer_output_meta_fn=["x"], is_subgraph

ProfilingContext4DebugL2("Run HiDream i2i input encoder")

ProfilingContext4DebugL2("Run HiDream input encoder")

magi_register_custom_op( "lightx2v::sage_attn2", infer_output_meta_fn=["q"], is_subgra

magi_register_custom_op( "lightx2v::triton_layer_norm", infer_output_meta_fn=["x"], is

router.post("/", response_model=TaskResponse)

router.post("/form", response_model=TaskResponse)

router.post("/sync")

router.post("/edits", response_model=OpenAIImageResponse)

router.post("/generations", response_model=OpenAIImageResponse)

router.post("/", response_model=TaskResponse)

router.post("/form", response_model=TaskResponse)

router.get("/download/{file_path:path}")

router.get("/queue/status")

router.get("/metadata")

router.get("/status")

router.get("/{task_id}/result")

router.get("/{task_id}/status")

self.app.get("/health")

magi_compile( dynamic_arg_dims={ "hidden_states": 0, "encoder_hidd

magi_compile( dynamic_arg_dims={ "hidden_states": 0, "encoder_hidd

router.get("/")

self.app.get("/")

router.delete("/all/running", response_model=StopTaskResponse)

router.delete("/{task_id}", response_model=StopTaskResponse)