Skip to content

Commit 33d8b52

Browse files
bconstantineAki-07
authored andcommitted
removed deprecated flag pin_first_last
1 parent ffad316 commit 33d8b52

File tree

1 file changed

+0
-4
lines changed

1 file changed

+0
-4
lines changed

src/diffusers/pipelines/pipeline_utils.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1343,7 +1343,6 @@ def enable_group_offload(
13431343
offload_to_disk_path: Optional[str] = None,
13441344
exclude_modules: Optional[Union[str, List[str]]] = None,
13451345
pin_groups: Optional[Union[str, Callable]] = None,
1346-
pin_first_last: bool = False,
13471346
) -> None:
13481347
r"""
13491348
Applies group offloading to the internal layers of a torch.nn.Module. To understand what group offloading is,
@@ -1407,8 +1406,6 @@ def enable_group_offload(
14071406
pin_groups (`\"first_last\"` | `\"all\"` | `Callable`, *optional*):
14081407
Optionally keep selected groups on the onload device permanently. See `ModelMixin.enable_group_offload`
14091408
for details.
1410-
pin_first_last (`bool`, *optional*, defaults to `False`):
1411-
Deprecated alias for `pin_groups=\"first_last\"`.
14121409
14131410
Example:
14141411
```python
@@ -1450,7 +1447,6 @@ def enable_group_offload(
14501447
"low_cpu_mem_usage": low_cpu_mem_usage,
14511448
"offload_to_disk_path": offload_to_disk_path,
14521449
"pin_groups": pin_groups,
1453-
"pin_first_last": pin_first_last,
14541450
}
14551451
for name, component in self.components.items():
14561452
if name not in exclude_modules and isinstance(component, torch.nn.Module):

0 commit comments

Comments
 (0)