more

Author: sayakpaul

src/diffusers/pipelines/pipeline_utils.py

@@ -1044,7 +1044,7 @@ def enable_model_cpu_offload(self, gpu_id: Optional[int] = None, device: Union[t
         device_mod = getattr(torch, device.type, None)
         if hasattr(device_mod, "empty_cache") and device_mod.is_available():
             device_mod.empty_cache()  # otherwise we don't see the memory savings (but they probably exist)
-
+        print("Empty cache called.")
         all_model_components = {k: v for k, v in self.components.items() if isinstance(v, torch.nn.Module)}
 
         self._all_hooks = []