er (str or os.PathLike, optional) —
The path to offload weights if device_map contains the value "disk".   offload_state_dict (bool, optional) —
If True, temporarily offloads the CPU state dict to the hard drive to avoid running out of CPU RAM if
the weight of the CPU state dict + the biggest shard of the checkpoint does not fit. Defaults to True
when there is some disk offload.   low_cpu_mem_usage (bool, optional, defaults to True if torch ver