From f6c2fb90f849255f1b5473be9a9d7ac8c052f774 Mon Sep 17 00:00:00 2001 From: Nadav Timor Date: Sun, 15 Sep 2024 20:13:56 +0000 Subject: [PATCH] enhance logging --- poc/dsi.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/poc/dsi.py b/poc/dsi.py index 74f29c0..5014cac 100644 --- a/poc/dsi.py +++ b/poc/dsi.py @@ -301,7 +301,7 @@ async def run(self) -> None: any_rejected: bool tok_ids, any_rejected = self.rejection_sampler(response, mask) print( - f"Manager: Updated tok_ids with response {response.id}. The new tok_ids are {self.tok_ids}" + f"Manager: Updated tok_ids with response {response.id}" ) tok_ids_padded = torch.full_like(self.tok_ids[0, mask], -1) tok_ids_padded[: len(tok_ids)] = tok_ids @@ -310,7 +310,7 @@ async def run(self) -> None: ) self.tok_ids[0, mask] = tok_ids_padded print( - f"Manager: Token ids after assignment: {self.tok_ids[0, mask]}" + f"Manager: Token ids after assignment: {self.tok_ids}" ) if any_rejected: print(f"Manager: Rejected response {response.id}")