fix black in pre-commit (#1940)

2024-11-07 15:42:47 -08:00
parent dca87ec348
commit c77c1e05ba
29 changed files with 641 additions and 508 deletions
--- a/scripts/playground/reference_hf.py
+++ b/scripts/playground/reference_hf.py
@@ -104,15 +104,9 @@ if __name__ == "__main__":
        default="TinyLlama/TinyLlama-1.1B-Chat-v0.4",
        # default="meta-llama/Llama-2-7b-chat-hf",
    )
-    parser.add_argument(
-        "--max-new-tokens",
-        type=int,
-        default=16)
+    parser.add_argument("--max-new-tokens", type=int, default=16)

-    parser.add_argument(
-        "--dtype",
-        type=str,
-        default="float16")
+    parser.add_argument("--dtype", type=str, default="float16")

    args = parser.parse_args()