update to use dtype/ torch_dtype

sugunav14 · sugunav14 · commit 1fd84496940a · 2026-02-06T19:31:48.000Z
Signed-off-by: Suguna Velury &lt;178320438+sugunav14@users.noreply.github.com&gt;
diff --git a/examples/gpt-oss/sft.py b/examples/gpt-oss/sft.py
@@ -72,7 +72,8 @@ def main(script_args, training_args, model_args, quant_args):
         "revision": model_args.model_revision,
         "trust_remote_code": model_args.trust_remote_code,
         "attn_implementation": model_args.attn_implementation,
-        "torch_dtype": model_args.torch_dtype,
+        "torch_dtype": getattr(model_args, "torch_dtype", None)
+        or getattr(model_args, "dtype", None),
         "use_cache": not training_args.gradient_checkpointing,
     }
 

Original file line number	Diff line number	Diff line change
`@@ -72,7 +72,8 @@ def main(script_args, training_args, model_args, quant_args):`
`72`	`72`	`"revision": model_args.model_revision,`
`73`	`73`	`"trust_remote_code": model_args.trust_remote_code,`
`74`	`74`	`"attn_implementation": model_args.attn_implementation,`
`75`		`- "torch_dtype": model_args.torch_dtype,`
	`75`	`+ "torch_dtype": getattr(model_args, "torch_dtype", None)`
	`76`	`+ or getattr(model_args, "dtype", None),`
`76`	`77`	`"use_cache": not training_args.gradient_checkpointing,`
`77`	`78`	`}`
`78`	`79`