PolyAI-pheme / constants.py
taras-sereda's picture
minimal set of files to run inference; pheme-small checkpoint
96ee597
raw
history blame
162 Bytes
"""Constants file.
Copyright PolyAI Limited.
"""
SPKR_EMB_SIZE = 512
PAD = 1024
SPKR_1 = 1025
SPKR_2 = 1026
BOS_TOKEN_ID = 0
PAD_TOKEN_ID = 0
EOS_TOKEN_ID = 2