import tokenzier

wwwjn · wwwjn · commit 9b97778abf36 · 2025-10-21T11:43:56.000-07:00
diff --git a/tests/unit_tests/test_dataset_flux.py b/tests/unit_tests/test_dataset_flux.py
@@ -12,6 +12,7 @@
 
 from torchtitan.config import ConfigManager
 from torchtitan.hf_datasets import DatasetConfig
+from torchtitan.models.flux.tokenizer import build_flux_tokenizer
 
 
 class TestFluxDataLoader(unittest.TestCase):
