Spaces:
Running
on
Zero
Running
on
Zero
File size: 1,099 Bytes
87c9ae9 25daeff 87c9ae9 25daeff 87c9ae9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 |
import argparse
import subprocess
import sys
from qwenimage.experiments.experiments_qwen import ExperimentRegistry
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--iterations", type=int, default=4)
parser.add_argument("--names", nargs="+", default=None)
args = parser.parse_args()
experiment_names = ExperimentRegistry.keys()
if args.names:
wrong_names = [name for name in args.names if name not in experiment_names]
if len(wrong_names) > 0:
raise ValueError(f"Names not in registry {wrong_names}")
else:
experiment_names = args.names
print(f"{len(experiment_names)}x {experiment_names}")
for name in experiment_names:
print(name)
cmd = [
sys.executable,
"scripts/run_experiment.py",
"--name", name,
"--iterations", str(args.iterations),
]
result = subprocess.run(cmd, check=True, capture_output=False, text=True)
print(result)
if __name__ == "__main__":
main()
|