iGPT/models/__init__.py

# from .image import (MaskFormer, ImageEditing, InstructPix2Pix, \
#     Text2Image, ImageCaptioning, Image2Canny, CannyText2Image, \
#     Image2Line, LineText2Image, Image2Hed, HedText2Image, Image2Scribble, \
#     ScribbleText2Image, Image2Pose, PoseText2Image, SegText2Image, \
#     Image2Depth, DepthText2Image, Image2Normal, NormalText2Image, \
#     VisualQuestionAnswering, InfinityOutPainting, \
#     SegmentAnything, InpaintMaskedAnything, ExtractMaskedAnything, \
#     ReplaceMaskedAnything, ImageOCRRecognition)

from .husky import HuskyVQA

from .video import (ActionRecognition, DenseCaption, VideoCaption, 
                    Summarization, GenerateTikTokVideo)

from .lang import SimpleLanguageModel

from .inpainting import LDMInpainting

# __all__ = [
#     'MaskFormer', 'ImageEditing', 'InstructPix2Pix', \
#     'Text2Image', 'ImageCaptioning', 'Image2Canny', 'CannyText2Image', \
#     'Image2Line', 'LineText2Image', 'Image2Hed', 'HedText2Image', \
#     'Image2Scribble', 'ScribbleText2Image', 'Image2Pose', 'PoseText2Image', \
#     'SegText2Image', 'Image2Depth', 'DepthText2Image', 'Image2Normal', \
#     'NormalText2Image', 'VisualQuestionAnswering', 'InfinityOutPainting', \
#     'SegmentAnything', 'InpaintMaskedAnything', 'ExtractMaskedAnything', \
#     'ReplaceMaskedAnything', 'ImageOCRRecognition', "SimpleLanguageModel", \
#     'ActionRecognition', 'DenseCaption', 'VideoCaption', 'Summarization', \
#     'GenerateTikTokVideo'
# ]

__all__ = [
    'HuskyVQA', "SimpleLanguageModel", 'GenerateTikTokVideo', \
    'LDMInpainting',
    'ActionRecognition', 'DenseCaption', 'VideoCaption', 'Summarization' 
]