Update flux_535m.py

Signed-off-by: Yu Yao <[email protected]>
NVIDIA · Feb 5, 2025 · 17b8d65 · 17b8d65
1 parent f2dc0b5
commit 17b8d65
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nemo/collections/diffusion/recipes/flux_535m.py b/nemo/collections/diffusion/recipes/flux_535m.py
@@ -25,7 +25,7 @@
 from nemo import lightning as nl
 from nemo.collections import llm
 from nemo.collections.diffusion.models.flux.model import FluxModelParams, MegatronFluxModel
-from nemo.collections.llm.recipes.log.default import default_log, default_resume, tensorboard_logger
+from nemo.collections.llm.recipes.log.default import default_resume, tensorboard_logger
 
 NAME = "flux-535m"