add a comment change in model.py to enable voc-contrained decoding

microsoft · Sep 28, 2023 · faae4fb · faae4fb
1 parent 187fc22
commit faae4fb
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/generativeimage2text/model.py b/generativeimage2text/model.py
@@ -39,6 +39,14 @@ def get_git_model(tokenizer, param):
         length_penalty=0.6,
     )
 
+    #from .trie_decoder import TrieAutoRegressiveBeamSearch, get_trie
+    #decoder = TrieAutoRegressiveBeamSearch(
+        #eos_index=tokenizer.sep_token_id,
+        #max_steps=1022,
+        #beam_size=1,
+        #trie=get_trie(tokenizer),
+    #)
+
     model = CaptioningModel(
         image_encoder,
         text_decoder,