Adjust the err

huggingface · Aug 19, 2024 · 1cb5649 · 1cb5649
1 parent 8c5b01d
commit 1cb5649
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/accelerate/inference.py b/src/accelerate/inference.py
@@ -154,7 +154,7 @@ def prepare_pippy(
             If `True`, the output from the last GPU (which holds the true outputs) is sent across to all GPUs.
     """
     if not is_pippy_available():
-        raise ImportError("Using PiPPy requires PyTorch 2.4.0 or later.")
+        raise ImportError("Using `torch.distributed.pipelining` requires PyTorch 2.4.0 or later.")
     state = PartialState()
     example_args = send_to_device(example_args, "cpu")
     example_kwargs = send_to_device(example_kwargs, "cpu")