- import os
- INITIAL_PEERS = os.environ.get("INITIAL_PEERS")
- if not INITIAL_PEERS:
- raise RuntimeError("Must specify INITIAL_PEERS environment variable with one or more peer ids")
- INITIAL_PEERS = INITIAL_PEERS.split()
- MODEL_NAME = os.environ.get("MODEL_NAME")
- if not MODEL_NAME:
- raise RuntimeError("Must specify MODEL_NAME as an index of a transformer block to be tested")
- REF_NAME = os.environ.get("REF_NAME")
- ADAPTER_NAME = os.environ.get("ADAPTER_NAME")
|