Skip to content

Commit ef95907

Browse files
[Tests] Relax tolerance of flaky failing test (#3755)
relax tolerance slightly
1 parent a812fb6 commit ef95907

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/models/test_models_unet_3d_condition.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -261,7 +261,7 @@ def test_lora_save_load(self):
261261
with torch.no_grad():
262262
new_sample = new_model(**inputs_dict, cross_attention_kwargs={"scale": 0.5}).sample
263263

264-
assert (sample - new_sample).abs().max() < 5e-4
264+
assert (sample - new_sample).abs().max() < 1e-3
265265

266266
# LoRA and no LoRA should NOT be the same
267267
assert (sample - old_sample).abs().max() > 1e-4

0 commit comments

Comments
 (0)