Skip to content

Commit 859c4a7

Browse files
huvunvidiaHuy Vu2github-actions[bot]
authored
Use EnergonMultiModalDataModule from M-bridge (#79)
* initial commit * update import EnergonMultiModalDataModule * update submodule Megatron-Bridge * Update uv.lock [skip ci] * update uv.lock * small update * small update --------- Co-authored-by: Huy Vu2 <[email protected]> Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
1 parent c807a8e commit 859c4a7

File tree

4 files changed

+40
-461
lines changed

4 files changed

+40
-461
lines changed

3rdparty/Megatron-Bridge

Submodule Megatron-Bridge updated 188 files

dfm/src/megatron/data/common/base_energon_datamodule.py

Lines changed: 0 additions & 362 deletions
This file was deleted.

dfm/src/megatron/data/common/diffusion_energon_datamodule.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,11 +19,11 @@
1919
from dataclasses import dataclass
2020
from typing import Any, Dict, Literal
2121

22+
from megatron.bridge.data.energon.base_energon_datamodule import EnergonMultiModalDataModule
2223
from megatron.bridge.data.utils import DatasetBuildContext, DatasetProvider
2324
from megatron.energon import DefaultTaskEncoder, get_train_dataset
2425
from torch import int_repr
2526

26-
from dfm.src.megatron.data.common.base_energon_datamodule import EnergonMultiModalDataModule
2727
from dfm.src.megatron.data.dit.dit_taskencoder import DiTTaskEncoder
2828

2929

0 commit comments

Comments
 (0)