Method register

diffbir/sampler/sampler.py:26–29

(
        self, name: str, value: np.ndarray, dtype: torch.dtype = torch.float32
    )

        self.rescale_cfg = rescale_cfg

    def register(
        self, name: str, value: np.ndarray, dtype: torch.dtype = torch.float32
    ) -> None:
        self.register_buffer(name, torch.tensor(value, dtype=dtype))

    def get_cfg_scale(self, default_cfg_scale: float, model_t: int) -> float:
        if self.rescale_cfg and default_cfg_scale > 1: