Method _set_context

tests/hooks/test_mag_cache.py:86–90

Helper to set context on all hooks in the model.

(self, model, context_name)

        )

    def _set_context(self, model, context_name):
        """Helper to set context on all hooks in the model."""
        for module in model.modules():
            if hasattr(module, "_diffusers_hook"):
                module._diffusers_hook._set_context(context_name)

    def _get_calibration_data(self, model):
        for module in model.modules():