[Bug] z_image_turbo error

[zhm-algo]

bash scripts/z_image/z_image_turbo_t2i.sh

will meet error shown as below

Traceback (most recent call last):
File "", line 198, in _run_module_as_main
File "", line 88, in _run_code
File "/root/LightX2V/lightx2v/infer.py", line 169, in
main()
File "/root/LightX2V/lightx2v/infer.py", line 160, in main
runner.run_pipeline(input_info)
File "/root/LightX2V/lightx2v/models/runners/z_image/z_image_runner.py", line 341, in run_pipeline
latents, generator = self.run_dit()
^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/utils/profiler.py", line 94, in sync_wrapper
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/runners/z_image/z_image_runner.py", line 84, in _run_dit_local
latents, generator = self.run(total_steps)
^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/runners/z_image/z_image_runner.py", line 224, in run
self.model.infer(self.inputs)
File "/root/miniforge3/envs/lightx2v/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/model.py", line 352, in infer
noise_pred = self._infer_cond_uncond(latents_input, inputs["text_encoder_output"]["prompt_embeds"], infer_condition=True)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/miniforge3/envs/lightx2v/lib/python3.11/site-packages/torch/utils/_contextlib.py", line 120, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/model.py", line 371, in _infer_cond_uncond
hidden_states = self.transformer_infer.infer(
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/infer/transformer_infer.py", line 284, in infer
hidden_states = self.infer_calculating(
^^^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/infer/transformer_infer.py", line 192, in infer_calculating
hidden_states = self.infer_block(
^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/infer/transformer_infer.py", line 124, in infer_block
attn_out = self.apply_attn(
^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/infer/transformer_infer.py", line 65, in apply_attn
query, key = self.apply_rope_func(query, key, freqs_cis_2d)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/root/LightX2V/lightx2v/models/networks/z_image/infer/utils.py", line 23, in apply_wan_rope_with_flashinfer
apply_rope_with_cos_sin_cache_inplace(
TypeError: 'NoneType' object is not callable