mirror of
https://github.com/ml-explore/mlx-examples.git
synced 2025-08-30 02:53:41 +08:00
t5: Change convert.py default to float32
This commit is contained in:
parent
3f91d8b9e8
commit
e39ad4cdf6
@ -71,7 +71,7 @@ if __name__ == "__main__":
|
||||
help="The model data type.",
|
||||
type=str,
|
||||
choices=["float16", "float32"],
|
||||
default="float16",
|
||||
default="float32",
|
||||
)
|
||||
args = parser.parse_args()
|
||||
convert(args.model, args.dtype)
|
||||
|
Loading…
Reference in New Issue
Block a user