pytorch
diff --git a/‎torchtitan/models/__init__.py‎
Lines changed: 1 addition & 1 deletion b/‎torchtitan/models/__init__.py‎
Lines changed: 1 addition & 1 deletion
@@ -5,5 +5,5 @@
 # LICENSE file in the root directory of this source tree.
 
 _supported_models = frozenset(
-    ["deepseek_v3", "flux", "gpt_oss", "llama3", "qwen3", "qwen3_5"]
+    ["deepseek_v3", "deepseek_v4", "flux", "gpt_oss", "llama3", "qwen3", "qwen3_5"]
 )
Original file line number	Diff line number	Diff line change
`@@ -5,5 +5,5 @@`
`5`	`5`	`# LICENSE file in the root directory of this source tree.`
`6`	`6`
`7`	`7`	`_supported_models = frozenset(`
`8`		`- ["deepseek_v3", "flux", "gpt_oss", "llama3", "qwen3", "qwen3_5"]`
	`8`	`+ ["deepseek_v3", "deepseek_v4", "flux", "gpt_oss", "llama3", "qwen3", "qwen3_5"]`
`9`	`9`	`)`