File size: 457 Bytes
1ab067e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 | import argparse
from nato1000_runtime.config import RuntimeConfig
from nato1000_runtime.runtime import NATO1000Runtime
parser = argparse.ArgumentParser(description="NATO-1000 runtime")
parser.add_argument("prompt")
parser.add_argument("--model-id", default="NATO1000-SENTINEL")
args = parser.parse_args()
cfg = RuntimeConfig(model_id=args.model_id)
cfg.validate()
runtime = NATO1000Runtime(cfg)
print(runtime.run(args.prompt).text)
print(runtime.health())
|