Merge branch 'hiyouga:main' into main

Former-commit-id: a4ce5ee381fd59f6b254ab634af51b6bb54edd97

src/llamafactory/hparams/model_args.py

@@ -149,7 +149,7 @@ class ModelArguments:
         default=1,
         metadata={"help": "The file shard size (in GB) of the exported model."},
     )
-    export_device: str = field(
+    export_device: Literal["cpu", "cuda"] = field(
         default="cpu",
         metadata={"help": "The device used in model export, use cuda to avoid addmm errors."},
     )