Skip to content

GitLab

Unverified Commit ef959071 authored by Patrick von Platen's avatar Patrick von Platen Committed by GitHub
Browse files

[Tests] Relax tolerance of flaky failing test (#3755) 

relax tolerance slightly
parent a812fb6f
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/test_models_unet_3d_condition.py
View file @ ef959071
...@@ -261,7 +261,7 @@ class UNet3DConditionModelTests(ModelTesterMixin, unittest.TestCase): ...@@ -261,7 +261,7 @@ class UNet3DConditionModelTests(ModelTesterMixin, unittest.TestCase):
with torch.no_grad(): with torch.no_grad():
new_sample = new_model(**inputs_dict, cross_attention_kwargs={"scale": 0.5}).sample new_sample = new_model(**inputs_dict, cross_attention_kwargs={"scale": 0.5}).sample
assert (sample - new_sample).abs().max() < 5e-4 assert (sample - new_sample).abs().max() < 1e-3
# LoRA and no LoRA should NOT be the same # LoRA and no LoRA should NOT be the same
assert (sample - old_sample).abs().max() > 1e-4 assert (sample - old_sample).abs().max() > 1e-4
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment