Update src/diffusers/pipelines/pipeline_utils.py

SunMarc · sayakpaul · web-flow · commit 54a903bedee6 · 2024-12-27T15:10:13.000+01:00
Co-authored-by: Sayak Paul &lt;spsayakpaul@gmail.com&gt;
diff --git a/src/diffusers/pipelines/pipeline_utils.py b/src/diffusers/pipelines/pipeline_utils.py
@@ -415,7 +415,7 @@ def module_is_offloaded(module):
         is_pipeline_device_mapped = self.hf_device_map is not None and len(self.hf_device_map) > 1
         if is_pipeline_device_mapped:
             raise ValueError(
-                "It seems like you have activated a device mapping strategy on the pipeline which doesn't allow explicit device placement using `to()`. You can call `reset_device_map()` first and then call `to()`."
+                "It seems like you have activated a device mapping strategy on the pipeline which doesn't allow explicit device placement using `to()`. You can call `reset_device_map()` to remove the existing device map from the pipeline."
             )
 
         if device and torch.device(device).type == "cuda":

Original file line number	Diff line number	Diff line change
`@@ -415,7 +415,7 @@ def module_is_offloaded(module):`
`415`	`415`	`is_pipeline_device_mapped = self.hf_device_map is not None and len(self.hf_device_map) > 1`
`416`	`416`	`if is_pipeline_device_mapped:`
`417`	`417`	`raise ValueError(`
`418`		- "It seems like you have activated a device mapping strategy on the pipeline which doesn't allow explicit device placement using `to()`. You can call `reset_device_map()` first and then call `to()`."
	`418`	+ "It seems like you have activated a device mapping strategy on the pipeline which doesn't allow explicit device placement using `to()`. You can call `reset_device_map()` to remove the existing device map from the pipeline."
`419`	`419`	`)`
`420`	`420`
`421`	`421`	`if device and torch.device(device).type == "cuda":`