AutoModel
class for image-text-to-text
models
#32042
Labels
Feature request
Request for a new feature
AutoModel
class for image-text-to-text
models
#32042
Feature request
It would be nice to get a standard
AutoModel
class forimage-text-to-text
models (since @molbap is standardizing the processor)Motivation
@NielsRogge noticed that in model repositories the automatic snippets fallback to
AutoModelForPreTraining
because these models don't exist inPIPELINE_TAGS_AND_AUTO_MODELS
(due to lack ofAutoClass
) More importantly it would be nice to load it to a single class.Your contribution
I haven't checked what it takes to implement an
AutoClass
when model classes exist in different names for the same task but if decided I don't mind looking into it and taking a stab.The text was updated successfully, but these errors were encountered: