Some functions are missing from the code

Hi @YanFangCS ,

I am trying to run the code you shared, but some of the functions are missing (probably due to the cleanup process):

In veomni/models/module_utils.py the GradientCheckpointingLayer is not present:

from ...module_utils import GradientCheckpointingLayer
  ImportError: cannot import name 'GradientCheckpointingLayer' from 'veomni.models.module_utils' 

These functions in veomni/data/__init__.py are not available as well:

from .dataset import (
    ...,
    build_iterative_vqa_webdataset,
    build_mixed_iterative_webdataset,
    ...
)

Start by inspecting veomni/models/module_utils.py and veomni/data/init.py, then run the reported imports to confirm which names are unavailable. Trace the corresponding dataset module and import paths; the work is done when GradientCheckpointingLayer and the two webdataset builders can be imported successfully without breaking existing imports.