diff --git a/xinfer/transformers/blip2.py b/xinfer/transformers/blip2.py index a223b7e..76304b5 100644 --- a/xinfer/transformers/blip2.py +++ b/xinfer/transformers/blip2.py @@ -32,6 +32,16 @@ "transformers", ModelInputOutput.IMAGE_TEXT_TO_TEXT, ) +@register_model( + "Gregor/mblip-mt0-xl", + "transformers", + ModelInputOutput.IMAGE_TEXT_TO_TEXT, +) +@register_model( + "Gregor/mblip-bloomz-7b", + "transformers", + ModelInputOutput.IMAGE_TEXT_TO_TEXT, +) class BLIP2(Vision2SeqModel): def __init__(self, model_id: str, **kwargs): super().__init__(model_id, **kwargs)