default to cogvlm

2024-07-04 13:07:48 -04:00 · 2024-07-04 13:07:48 -04:00 · bc1e3e7209
parent e507065df2
commit bc1e3e7209
1 changed files with 3 additions and 0 deletions
--- a/caption.py
+++ b/caption.py
@ -485,6 +485,9 @@ class CogVLMManager(BaseModelWrapper):
        return caption

 def get_model_wrapper(model_name: str):
+    if model_name is None:
+        logging.info("No model given, defaulting to CogVLM")
+        return CogVLMManager("thudm/cogvlm-chat-hf")
    match model_name.casefold():
        # case x if "moai" in x:
        #     #return MoaiManager(model_name)