Spaces:

umichVision
/

virtex-redcaps

Runtime error

Black + isort, remove unused virtx files.

8d0e872 over 2 years ago

No virus

515 Bytes

	from .captioning import (
	ForwardCaptioningModel,
	BidirectionalCaptioningModel,
	VirTexModel
	)
	from .masked_lm import MaskedLMModel
	from .classification import (
	MultiLabelClassificationModel,
	TokenClassificationModel,
	)
	from .contrastive import ImageTextContrastiveModel


	__all__ = [
	"VirTexModel",
	"BidirectionalCaptioningModel",
	"ForwardCaptioningModel",
	"MaskedLMModel",
	"MultiLabelClassificationModel",
	"TokenClassificationModel",
	"ImageTextContrastiveModel",
	]