You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
14 lines
419 B
Python
14 lines
419 B
Python
2 years ago
|
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")
|