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)