Skip to content

Commit 737904a

Browse files
Fix import bug
1 parent c98f700 commit 737904a

File tree

1 file changed

+1
-0
lines changed
  • src/monitors4codegen/monitor_guided_decoding

1 file changed

+1
-0
lines changed

src/monitors4codegen/monitor_guided_decoding/hf_gen.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -8,6 +8,7 @@
88
from asyncio.events import AbstractEventLoop
99
from typing import List, Union
1010
from transformers import LogitsProcessor
11+
from monitors4codegen.monitor_guided_decoding.monitor import Monitor
1112

1213
class MGDLogitsProcessor(LogitsProcessor):
1314
"""

0 commit comments

Comments
 (0)