@@ -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