huggingface/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-05-28 00:39:35 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

[tests] Add fast test slices for HiDream-Image (#11953)
Some checks are pending
Run dependency tests / check_dependencies (push) Waiting to run
Details
Run Flax dependency tests / check_flax_dependencies (push) Waiting to run
Details
Run Torch dependency tests / check_torch_dependencies (push) Waiting to run
Details
Fast GPU Tests on main / Setup Torch Pipelines CUDA Slow Tests Matrix (push) Waiting to run
Details
Fast GPU Tests on main / Torch Pipelines CUDA Tests (push) Blocked by required conditions
Details
Fast GPU Tests on main / Torch CUDA Tests (lora) (push) Waiting to run
Details
Fast GPU Tests on main / Torch CUDA Tests (models) (push) Waiting to run
Details
Fast GPU Tests on main / Torch CUDA Tests (others) (push) Waiting to run
Details
Fast GPU Tests on main / Torch CUDA Tests (schedulers) (push) Waiting to run
Details
Fast GPU Tests on main / Torch CUDA Tests (single_file) (push) Waiting to run
Details
Fast GPU Tests on main / PyTorch Compile CUDA tests (push) Waiting to run
Details
Fast GPU Tests on main / PyTorch xformers CUDA tests (push) Waiting to run
Details
Fast GPU Tests on main / Examples PyTorch CUDA tests on Ubuntu (push) Waiting to run
Details
Fast tests on main / ${{ matrix.config.name }} (map[framework:pytorch image:diffusers/diffusers-pytorch-cpu name:Fast PyTorch CPU tests on Ubuntu report:torch_cpu runner:aws-general-8-plus]) (push) Waiting to run
Details
Fast tests on main / ${{ matrix.config.name }} (map[framework:pytorch_examples image:diffusers/diffusers-pytorch-cpu name:PyTorch Example CPU tests on Ubuntu report:torch_example_cpu runner:aws-general-8-plus]) (push) Waiting to run
Details
Secret Leaks / trufflehog (push) Waiting to run
Details
Update Diffusers metadata / update_metadata (push) Waiting to run
Details

Browse Source 
update
This commit is contained in:
Aryan
2025-07-21 07:53:13 +05:30
committed by GitHub
parent d87134ada4
commit 9db9be65f3
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/pipelines/hidream_image/test_pipeline_hidream.py
View File

@@ -146,11 +146,15 @@ class HiDreamImagePipelineFastTests(PipelineTesterMixin, unittest.TestCase):
inputs = self.get_dummy_inputs(device)
image = pipe(**inputs)[0]
generated_image = image[0]
self.assertEqual(generated_image.shape, (128, 128, 3))
expected_image = torch.randn(128, 128, 3).numpy()
max_diff = np.abs(generated_image - expected_image).max()
self.assertLessEqual(max_diff, 1e10)
# fmt: off
expected_slice = np.array([0.4507, 0.5256, 0.4205, 0.5791, 0.4848, 0.4831, 0.4443, 0.5107, 0.6586, 0.3163, 0.7318, 0.5933, 0.6252, 0.5512, 0.5357, 0.5983])
# fmt: on
generated_slice = generated_image.flatten()
generated_slice = np.concatenate([generated_slice[:8], generated_slice[-8:]])
self.assertTrue(np.allclose(generated_slice, expected_slice, atol=1e-3))
def test_inference_batch_single_identical(self):
super().test_inference_batch_single_identical(expected_max_diff=3e-4)