Fix import bug

microsoft · Nov 21, 2023 · 737904a · 737904a
1 parent c98f700
commit 737904a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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):
     """