Spaces:
Build error
Build error
File size: 902 Bytes
844e150 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from .open_flamingo import create_model_and_transforms as create_open_flamingo_model_and_transforms
def create_model_and_transforms(
model_name: str,
clip_vision_encoder_path: str,
clip_vision_encoder_pretrained: str,
lang_encoder_path: str,
tokenizer_path: str,
tuning_config,
pretrained_model_path,
**kwargs,
):
if model_name == "open_flamingo":
return create_open_flamingo_model_and_transforms(
clip_vision_encoder_path=clip_vision_encoder_path,
clip_vision_encoder_pretrained=clip_vision_encoder_pretrained,
lang_encoder_path=lang_encoder_path,
tokenizer_path=tokenizer_path,
tuning_config=tuning_config,
pretrained_model_path=pretrained_model_path,
**kwargs,
)
# TODO: support BLIP2
else:
raise ValueError(f"Unknown model name: {model_name}")
|