Skip to content

skip invalid test case for helios pipeline#13218

Merged
yiyixuxu merged 4 commits intohuggingface:mainfrom
kaixuanliu:helio-fix
Mar 18, 2026
Merged

skip invalid test case for helios pipeline#13218
yiyixuxu merged 4 commits intohuggingface:mainfrom
kaixuanliu:helio-fix

Conversation

@kaixuanliu
Copy link
Contributor

In helios pipeline, the latents are always in FP32 format: L798, hence we should skip the test_save_load_float16 test case, rather than adjust the tolerance.

Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
@kaixuanliu
Copy link
Contributor Author

@DN6 pls help review, thx!

@kaixuanliu
Copy link
Contributor Author

@yiyixuxu Hi, can you help review? Thx!

@yiyixuxu
Copy link
Collaborator

hey @kaixuanliu
latents are initially created in float32 and processed in the actual dtype of transformere though https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/helios/pipeline_helios.py#L817

did I miss anything?

Signed-off-by: Liu, Kaixuan <kaixuan.liu@intel.com>
@kaixuanliu
Copy link
Contributor Author

@yiyixuxu Thx for the comment! Yes, I am afraid you are right. But for this pipeline, it has a lot of modules using mixed precision, such as: keep_in_fp32_module, FP32LayerNorm, L435-L437 and HeliosOutputNorm, hence I suggest to skip this test case like in WanVACE pipeline, and I have updated the skip reason. Do you think it is OK?

Copy link
Collaborator

@yiyixuxu yiyixuxu left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thanks!

@yiyixuxu yiyixuxu merged commit 8e4b560 into huggingface:main Mar 18, 2026
7 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants