From 7c1298ccc480b42355baf957f3aed0eeb7432d87 Mon Sep 17 00:00:00 2001 From: Dan Saattrup Nielsen Date: Wed, 22 May 2024 13:09:30 +0200 Subject: [PATCH] debug --- src/ragger/generator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/ragger/generator.py b/src/ragger/generator.py index 281610c4..618dd5ff 100644 --- a/src/ragger/generator.py +++ b/src/ragger/generator.py @@ -208,6 +208,7 @@ def __init__(self, config: DictConfig) -> None: # Load the tokenizer without printing any logs sys.stderr = open(os.devnull, "w") + breakpoint() self.tokenizer = AutoTokenizer.from_pretrained(config.generator.model) sys.stderr = sys.__stderr__