Skip to content

Commit 6d96002

Browse files
committed
removed pull request code comments
1 parent 9950689 commit 6d96002

File tree

1 file changed

+0
-3
lines changed

1 file changed

+0
-3
lines changed

tests/hooks/test_group_offloading.py

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,6 @@ def _normalize_module_type(self, obj: Any) -> Optional[torch.nn.Module]:
182182
return mods[0] if len(mods) == 1 else None
183183
return None
184184

185-
# Test for https://github.com/huggingface/diffusers/pull/12747
186185
class DummyCallableByNameSubmodule(DummyCallableBySubmodule):
187186
"""
188187
Callable group offloading pinner that pins first and last DummyBlock
@@ -193,7 +192,6 @@ def __call__(self, name: str, submodule: torch.nn.Module) -> bool:
193192
self.calls_track.append((name, submodule))
194193
return self._normalize_module_type(submodule) in self.pin_targets
195194

196-
# Test for https://github.com/huggingface/diffusers/pull/12747
197195
class DummyCallableByNameSubmoduleIdx(DummyCallableBySubmodule):
198196
"""
199197
Callable group offloading pinner that pins first and last DummyBlock.
@@ -204,7 +202,6 @@ def __call__(self, name: str, submodule: torch.nn.Module, idx: int) -> bool:
204202
self.calls_track.append((name, submodule, idx))
205203
return self._normalize_module_type(submodule) in self.pin_targets
206204

207-
# Test for https://github.com/huggingface/diffusers/pull/12747
208205
class DummyInvalidCallable(DummyCallableBySubmodule):
209206
"""
210207
Callable group offloading pinner that uses invalid call signature

0 commit comments

Comments
 (0)