"deploy/operator/vscode:/vscode.git/clone" did not exist on "a96d9c513f63f2f6345583e9cb0af9f5b408bbbd"
Unverified Commit 1dd076cc authored by Qi Wang's avatar Qi Wang Committed by GitHub
Browse files

feat(bench): add seed support to multimodal sweep and JSONL generator (#7995)


Co-authored-by: default avatarClaude Opus 4.6 (1M context) <noreply@anthropic.com>
parent 66f7832a
......@@ -78,4 +78,10 @@ def parse_args(description: str = "") -> argparse.Namespace:
metavar=("WIDTH", "HEIGHT"),
help="Size of generated PNG images in pixels (default: 512 512)",
)
parser.add_argument(
"--seed",
type=int,
default=None,
help="Random seed for reproducible dataset generation (default: time-based)",
)
return parser.parse_args()
......@@ -26,8 +26,6 @@ from generate_images import (
)
from generate_input_text import generate_filler
SEED = int(time.time() * 1000) % (2**32)
def main() -> None:
args = parse_args(__doc__)
......@@ -35,8 +33,13 @@ def main() -> None:
images_per_request: int = args.images_per_request
image_pool: int = args.images_pool or (num_requests * images_per_request)
np_rng = np.random.default_rng(SEED)
py_rng = random.Random(SEED)
seed: int = (
args.seed if args.seed is not None else int(time.time() * 1000) % (2**32)
)
print(f"Using seed: {seed}")
np_rng = np.random.default_rng(seed)
py_rng = random.Random(seed)
if args.image_mode == "http":
pool = generate_image_pool_http(py_rng, image_pool, args.coco_annotations)
......
......@@ -70,5 +70,4 @@ def parse_args(argv=None) -> argparse.Namespace:
default=None,
help="Skip plot generation.",
)
return parser.parse_args(argv)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment