use fixture

Former-commit-id: 80a9e6bf94
This commit is contained in:
hiyouga
2024-06-15 20:06:17 +08:00
parent 8053929b20
commit 96b82ccd4d
3 changed files with 24 additions and 7 deletions

View File

@@ -163,7 +163,7 @@ class ModelArguments:
)
infer_dtype: Literal["auto", "float16", "bfloat16", "float32"] = field(
default="auto",
metadata={"help": "Data type for model weights and activations at inference."}
metadata={"help": "Data type for model weights and activations at inference."},
)
hf_hub_token: Optional[str] = field(
default=None,