Skip to content

Commit 08211f7

Browse files
committed
skip test in NCSNppModelTests
1 parent 2c9c33f commit 08211f7

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

tests/models/unets/test_models_unet_2d.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -401,3 +401,15 @@ def test_gradient_checkpointing_is_applied(self):
401401

402402
def test_effective_gradient_checkpointing(self):
403403
super().test_effective_gradient_checkpointing(skip={"time_proj.weight"})
404+
405+
@unittest.skip(
406+
"To make layerwise upcasting work with this model, we will have to update the implementation. Due to potentially low usage, we don't support it here."
407+
)
408+
def test_layerwise_upcasting_inference(self):
409+
pass
410+
411+
@unittest.skip(
412+
"To make layerwise upcasting work with this model, we will have to update the implementation. Due to potentially low usage, we don't support it here."
413+
)
414+
def test_layerwise_upcasting_memory(self):
415+
pass

0 commit comments

Comments
 (0)