fix

allenai · Dec 5, 2023 · ac01778 · ac01778
1 parent 6c79c63
commit ac01778
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/unshard.py b/scripts/unshard.py
@@ -88,5 +88,5 @@ def main(
         args.input_dir,
         args.output_dir,
         sharded_checkpoint_type=args.type,
-        model_only=model_only,
+        model_only=args.model_only,
     )