Method reset_parameters

lit_gpt/model.py:113–118

(self)

                )

    def reset_parameters(self) -> None:
        # Trigger resetting the rope-cache
        try:
            self.cos, self.sin = self.get_rope_cache(device=self.cos.device)
        except Exception as e:
            print(e)

    def reset_cache(self) -> None:
        self.kv_caches.clear()