AI/diffusers
1
0
Fork 0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
Code Activity

Fix failing np tests (#3942)

Browse Source 
* Fix failing np tests

* Apply suggestions from code review

* Update tests/pipelines/test_pipelines_common.py
This commit is contained in:
Patrick von Platen
2023-07-04 14:00:43 +02:00
committed by GitHub
parent 1997614aa9
commit 2837d49079
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/pipelines/test_pipelines_common.py
View File

@@ -698,11 +698,13 @@ class PipelineTesterMixin:
pipe.set_progress_bar_config(disable=None)
inputs = self.get_dummy_inputs(torch_device)
output_without_offload = pipe(**inputs)[0].cpu()
output_without_offload = pipe(**inputs)[0]
output_without_offload.cpu() if torch.is_tensor(output_without_offload) else output_without_offload
pipe.enable_xformers_memory_efficient_attention()
inputs = self.get_dummy_inputs(torch_device)
output_with_offload = pipe(**inputs)[0].cpu()
output_with_offload = pipe(**inputs)[0]
output_with_offload.cpu() if torch.is_tensor(output_with_offload) else output_without_offload
if test_max_difference:
max_diff = np.abs(output_with_offload - output_without_offload).max()