set the default value of torch_dtype to None

ChaoPang · ChaoPang · commit 54159f9ad2e8 · 2025-04-10T22:36:00.000-04:00
diff --git a/src/cehrbert/runners/hf_runner_argument_dataclass.py b/src/cehrbert/runners/hf_runner_argument_dataclass.py
@@ -228,7 +228,7 @@ class ModelArguments:
         },
     )
     torch_dtype: Optional[str] = dataclasses.field(
-        default="auto",
+        default=None,
         metadata={
             "help": (
                 "Override the default `torch.dtype` and load the model under this dtype. If `auto` is passed, the "

Original file line number	Diff line number	Diff line change
`@@ -228,7 +228,7 @@ class ModelArguments:`
`228`	`228`	`},`
`229`	`229`	`)`
`230`	`230`	`torch_dtype: Optional[str] = dataclasses.field(`
`231`		`- default="auto",`
	`231`	`+ default=None,`
`232`	`232`	`metadata={`
`233`	`233`	`"help": (`
`234`	`234`	"Override the default `torch.dtype` and load the model under this dtype. If `auto` is passed, the "