@aws/ml-container-creator 0.13.5 → 0.15.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/config/parameter-schema-v2.json +32 -4
- package/infra/ci-harness/lib/ci-harness-stack.ts +13 -5
- package/infra/ci-harness/package-lock.json +121 -111
- package/infra/ci-harness/package.json +1 -1
- package/package.json +2 -2
- package/servers/instance-sizer/index.js +72 -4
- package/servers/instance-sizer/lib/model-resolver.js +28 -2
- package/src/app.js +15 -0
- package/src/lib/config-loader.js +18 -0
- package/src/lib/config-manager.js +6 -1
- package/src/lib/dataset-slug.js +152 -0
- package/src/lib/generated/cli-options.js +9 -3
- package/src/lib/generated/parameter-matrix.js +14 -3
- package/src/lib/generated/validation-rules.js +1 -1
- package/src/lib/mcp-query-runner.js +6 -0
- package/src/lib/prompt-runner.js +5 -0
- package/src/lib/prompts/feature-prompts.js +1 -1
- package/src/lib/template-manager.js +0 -7
- package/src/lib/template-variable-resolver.js +51 -1
- package/src/lib/tune-config-state.js +14 -1
- package/templates/do/.benchmark_writer.py +9 -0
- package/templates/do/.register_helper.py +1163 -0
- package/templates/do/.tune_helper.py +168 -2
- package/templates/do/__pycache__/.adapter_helper.cpython-312.pyc +0 -0
- package/templates/do/__pycache__/.benchmark_writer.cpython-312.pyc +0 -0
- package/templates/do/__pycache__/.register_helper.cpython-312.pyc +0 -0
- package/templates/do/__pycache__/.tune_helper.cpython-312.pyc +0 -0
- package/templates/do/adapter +319 -27
- package/templates/do/add-ic +85 -3
- package/templates/do/benchmark +28 -8
- package/templates/do/config +20 -0
- package/templates/do/lib/inference-component.sh +56 -3
- package/templates/do/register +552 -6
- package/templates/do/test +12 -2
- package/templates/do/tune +201 -6
|
@@ -487,6 +487,7 @@ def enrich_records(config, results, run_timestamp=None):
|
|
|
487
487
|
'mcc_version': mcc_version,
|
|
488
488
|
'run_timestamp': run_timestamp.isoformat(),
|
|
489
489
|
'region': region,
|
|
490
|
+
'adapter_name': config.get('adapter_name', ''),
|
|
490
491
|
}
|
|
491
492
|
records.append(record)
|
|
492
493
|
|
|
@@ -859,6 +860,7 @@ def get_parquet_schema():
|
|
|
859
860
|
pa.field("mcc_version", pa.string()),
|
|
860
861
|
pa.field("run_timestamp", pa.string()),
|
|
861
862
|
pa.field("region", pa.string()),
|
|
863
|
+
pa.field("adapter_name", pa.string()),
|
|
862
864
|
])
|
|
863
865
|
|
|
864
866
|
|
|
@@ -1177,6 +1179,8 @@ def cmd_write(args):
|
|
|
1177
1179
|
input_data['workload'] = args.workload
|
|
1178
1180
|
if args.region:
|
|
1179
1181
|
input_data['region'] = args.region
|
|
1182
|
+
if args.adapter_name:
|
|
1183
|
+
input_data['adapter_name'] = args.adapter_name
|
|
1180
1184
|
|
|
1181
1185
|
# ── Validate before any S3 interaction ────────────────────────────────
|
|
1182
1186
|
errors = validate_benchmark_input(input_data)
|
|
@@ -1462,6 +1466,11 @@ def main():
|
|
|
1462
1466
|
'--region',
|
|
1463
1467
|
help='AWS region'
|
|
1464
1468
|
)
|
|
1469
|
+
write_parser.add_argument(
|
|
1470
|
+
'--adapter-name', dest='adapter_name', default=None,
|
|
1471
|
+
help='LoRA adapter name (differentiates adapter benchmarks from base model in Athena)'
|
|
1472
|
+
)
|
|
1473
|
+
|
|
1465
1474
|
write_parser.add_argument(
|
|
1466
1475
|
'--dry-run', dest='dry_run', action='store_true',
|
|
1467
1476
|
help='Output enriched records as JSON without writing to S3'
|