Mirrors
/
EveryDream2trainer
mirror of https://github.com/victorchall/EveryDream2trainer.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

default to cogvlm

This commit is contained in:
Victor Hall 2024-07-04 13:07:48 -04:00
parent e507065df2
commit bc1e3e7209
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
caption.py
View File

@ -485,6 +485,9 @@ class CogVLMManager(BaseModelWrapper):
return caption return caption
def get_model_wrapper(model_name: str): 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(): match model_name.casefold():
# case x if "moai" in x: # case x if "moai" in x:
# #return MoaiManager(model_name) # #return MoaiManager(model_name)