refactor flux

wwwjn · wwwjn · commit bd52d3c5058f · 2025-10-18T21:30:54.000-07:00
diff --git a/.ci/docker/requirements-dev.txt b/.ci/docker/requirements-dev.txt
@@ -3,3 +3,4 @@ pytest==7.3.2
 pytest-cov
 pre-commit
 tomli-w >= 1.1.0
+transformers
diff --git a/.ci/docker/requirements-flux.txt b/.ci/docker/requirements-flux.txt
@@ -1,4 +1,4 @@
-transformers
+transformers>=4.51.1
 einops
 sentencepiece
 pillow
diff --git a/torchtitan/models/flux/requirements-flux.txt b/torchtitan/models/flux/requirements-flux.txt
@@ -1 +1 @@
-.ci/docker/requirements-flux.txt
+../../../.ci/docker/requirements-flux.txt

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-.ci/docker/requirements-flux.txt`
	`1`	`+../../../.ci/docker/requirements-flux.txt`