From 737904afc5b75b43d94c3340fcacd8ddef8c5006 Mon Sep 17 00:00:00 2001 From: Lakshya A Agrawal Date: Mon, 20 Nov 2023 08:48:12 +0000 Subject: [PATCH] Fix import bug --- src/monitors4codegen/monitor_guided_decoding/hf_gen.py | 1 + 1 file changed, 1 insertion(+) diff --git a/src/monitors4codegen/monitor_guided_decoding/hf_gen.py b/src/monitors4codegen/monitor_guided_decoding/hf_gen.py index 56d5564..2ab5568 100644 --- a/src/monitors4codegen/monitor_guided_decoding/hf_gen.py +++ b/src/monitors4codegen/monitor_guided_decoding/hf_gen.py @@ -8,6 +8,7 @@ from asyncio.events import AbstractEventLoop from typing import List, Union from transformers import LogitsProcessor +from monitors4codegen.monitor_guided_decoding.monitor import Monitor class MGDLogitsProcessor(LogitsProcessor): """