From bc1e3e72098451ece09b374e34632e02d75eb1fd Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Thu, 4 Jul 2024 13:07:48 -0400 Subject: [PATCH] default to cogvlm --- caption.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/caption.py b/caption.py index bc573c1..69a584a 100644 --- 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)