From 74163dfef1ddd9cbbfa3cd7ac00b7fc0129697b7 Mon Sep 17 00:00:00 2001 From: Gabriele Oliaro Date: Sun, 14 Jan 2024 21:11:56 +0000 Subject: [PATCH] fix --- inference/utils/download_hf_model.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/inference/utils/download_hf_model.py b/inference/utils/download_hf_model.py index 03fc8e1633..94a8c23e68 100644 --- a/inference/utils/download_hf_model.py +++ b/inference/utils/download_hf_model.py @@ -36,9 +36,9 @@ def parse_args(): def main(args): if args.full_precision_only: - data_types = ff.DataType.DT_FLOAT + data_types = (ff.DataType.DT_FLOAT,) elif args.half_precision_only: - data_types = ff.DataType.DT_HALF + data_types = (ff.DataType.DT_HALF,) else: data_types = (ff.DataType.DT_FLOAT, ff.DataType.DT_HALF)