Some functions are missing from the code

[kumuji]

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,
    ...
)

[YanFangCS]

Hi @kumuji,

So sorry for the late reply.

Thanks a lot for pointing this out. These missing functions were likely caused by some legacy issues during the code cleanup process. The latest version of our codebase should have properly addressed these previously remaining problems.

We will continue to check the simplified codebase and fix any remaining issues that may still exist. Please feel free to pull the latest code and let us know if you encounter any further problems.

Thanks again for your feedback!