Spaces:

PolyAI
/

pheme

Runtime error

App Files Files Community

pheme / constants.py

taras-sereda's picture

minimal set of files to run inference; pheme-small checkpoint

96ee597 about 1 year ago

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