Method _get_calibration_data

tests/hooks/test_mag_cache.py:92–98

(self, model)

                module._diffusers_hook._set_context(context_name)

    def _get_calibration_data(self, model):
        for module in model.modules():
            if hasattr(module, "_diffusers_hook"):
                hook = module._diffusers_hook.get_hook("mag_cache_block_hook")
                if hook:
                    return hook.state_manager.get_state().calibration_ratios
        return []

    def test_mag_cache_validation(self):
        """Test that missing mag_ratios raises ValueError."""