Freeze code long context profile v6
This commit is contained in:
@@ -46,6 +46,10 @@ def parse_args() -> argparse.Namespace:
|
||||
parser.add_argument("--output", type=Path, required=True,
|
||||
help="profile-v5-kvgrowth output directory")
|
||||
parser.add_argument("--max-model-len", type=int, default=40960)
|
||||
parser.add_argument(
|
||||
"--manifest-schema",
|
||||
default="frontier-profile-v5-kvgrowth-v1",
|
||||
)
|
||||
return parser.parse_args()
|
||||
|
||||
|
||||
@@ -221,7 +225,7 @@ def main() -> None:
|
||||
shutil.copy2(args.base / name, args.output / name)
|
||||
|
||||
manifest = {
|
||||
"schema": "frontier-profile-v5-kvgrowth-v1",
|
||||
"schema": args.manifest_schema,
|
||||
"base": str(base_csv),
|
||||
"base_sha256": sha256(base_csv),
|
||||
"raw_inputs": {str(p): sha256(p) for p in args.raw},
|
||||
|
||||
Reference in New Issue
Block a user