Get moondream tokenizer correctly

jhc13 · jhc13 · commit b980839041fa · 2024-03-12T21:37:44.000+09:00
diff --git a/taggui/auto_captioning/captioning_thread.py b/taggui/auto_captioning/captioning_thread.py
@@ -31,7 +31,7 @@
 
 def get_tokenizer_from_processor(model_type: ModelType, processor):
     if model_type in (ModelType.COGAGENT, ModelType.COGVLM,
-                      ModelType.XCOMPOSER2):
+                      ModelType.MOONDREAM, ModelType.XCOMPOSER2):
         return processor
     return processor.tokenizer
 
