Skip to content
Permalink

Comparing changes

This is a direct comparison between two commits made in this repository or its related repositories. View the default comparison for this range or learn more about diff comparisons.

Open a pull request

Create a new pull request by comparing changes across two branches. If you need to, you can also . Learn more about diff comparisons here.
base repository: mosaicml/composer
Failed to load repositories. Confirm that selected base ref is valid, then try again.
Loading
base: da96c16a2bfed581142f485d2adabb326f20ff46
Choose a base ref
..
head repository: mosaicml/composer
Failed to load repositories. Confirm that selected head ref is valid, then try again.
Loading
compare: fbff704b9c8e2ac81f98a661219c3c88dfcb9de9
Choose a head ref
Showing with 4 additions and 1 deletion.
  1. +4 −1 composer/utils/checkpoint.py
5 changes: 4 additions & 1 deletion composer/utils/checkpoint.py
Original file line number Diff line number Diff line change
@@ -595,7 +595,10 @@ def dist_cp_load(
storage_reader: StorageReader,
load_planner: Optional[LoadPlanner] = None,
):
if (version.parse(torch.__version__) >= version.parse('2.4.0') and version.parse(torch.__version__) < version.parse('2.5.0')):
if (
version.parse(torch.__version__) >= version.parse('2.4.0') and
version.parse(torch.__version__) < version.parse('2.5.0')
):
from torch.distributed.checkpoint.utils import CheckpointException
try:
dist_cp.load(