import os import subprocess import sys from pathlib import Path model = ('base', 'small')[int(os.environ['LOCAL_RANK'])] subprocess.run([sys.executable, str(Path(__file__).with_name('export_and_verify.py')), '--model', model], check=True)