Skip to content

Commit 91aab44

Browse files
committed
format
Signed-off-by: Ao Tang <aot@nvidia.com>
1 parent 1b8f660 commit 91aab44

2 files changed

Lines changed: 1 addition & 2 deletions

File tree

‎src/megatron/bridge/diffusion/recipes/flux/__init__.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -13,8 +13,8 @@
1313
# limitations under the License.
1414

1515
from megatron.bridge.diffusion.recipes.flux.flux import (
16-
flux_12b_sft_config,
1716
flux_12b_pretrain_config,
17+
flux_12b_sft_config,
1818
)
1919

2020

‎tests/unit_tests/diffusion/model/flux/conversion/test_flux_hf_pretrained.py‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -97,7 +97,6 @@ def test_save_artifacts_copies_existing_files(tmp_path):
9797
assert json.loads((dest_tdir / "diffusion_pytorch_model.safetensors.index.json").read_text()) == index_data
9898

9999

100-
101100
def test_save_artifacts_handles_export_failure(monkeypatch, tmp_path):
102101
class FailingFlux:
103102
@classmethod

0 commit comments

Comments
 (0)