From 82302262caa3ece3049ff6f061aa155bc64ab4ea Mon Sep 17 00:00:00 2001 From: OlivierDehaene <23298448+OlivierDehaene@users.noreply.github.com> Date: Wed, 12 Jun 2024 17:50:53 +0200 Subject: [PATCH] remove logs --- server/text_generation_server/layers/rotary.py | 3 --- 1 file changed, 3 deletions(-) diff --git a/server/text_generation_server/layers/rotary.py b/server/text_generation_server/layers/rotary.py index 5d92ad5e..c2f12189 100644 --- a/server/text_generation_server/layers/rotary.py +++ b/server/text_generation_server/layers/rotary.py @@ -280,9 +280,6 @@ class SuRotaryEmbedding(PositionRotaryEmbedding): freqs = torch.cat([short_freqs, long_freqs]) - from loguru import logger - - logger.info(freqs.shape) self._cos_cached = (torch.cos(freqs) * self.scaling_factor).to(dtype) self._sin_cached = (torch.sin(freqs) * self.scaling_factor).to(dtype)