fix

flexflow · Jan 14, 2024 · 74163df · 74163df
1 parent 621c20c
commit 74163df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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)