mirror of
https://github.com/bigscience-workshop/petals
synced 2024-10-31 09:20:41 +00:00
14 lines
419 B
Python
14 lines
419 B
Python
|
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")
|