Commit 1805450

committed

1 parent a3d205c commit 1805450Copy full SHA for 1805450

File tree

-0

lines changed

-0

lines changed

Lines changed: 5 additions & 0 deletions

Original file line number	Diff line number	Diff line change
`@@ -57,6 +57,9 @@ def test_enable_disable_tiling(self):`
`57`	`57`	`torch.manual_seed(0)`
`58`	`58`	`model = self.model_class(**init_dict).to(torch_device)`
`59`	`59`
	`60`	`+ if not hasattr(model, "use_tiling"):`
	`61`	`+ pytest.skip(f"Skipping test as {self.model_class.__name__} doesn't support tiling.")`
	`62`	`+`
`60`	`63`	`inputs_dict.update({"return_dict": False})`
`61`	`64`	`_ = inputs_dict.pop("generator", None)`
`62`	`65`	`accepts_generator = self._accepts_generator(model)`
`@@ -102,6 +105,8 @@ def test_enable_disable_slicing(self):`
`102`	`105`
`103`	`106`	`torch.manual_seed(0)`
`104`	`107`	`model = self.model_class(**init_dict).to(torch_device)`
	`108`	`+ if not hasattr(model, "use_slicing"):`
	`109`	`+ pytest.skip(f"Skipping test as {self.model_class.__name__} doesn't support tiling.")`
`105`	`110`
`106`	`111`	`inputs_dict.update({"return_dict": False})`
`107`	`112`	`_ = inputs_dict.pop("generator", None)`

Comments

(0)