16 lines
664 B
Python
16 lines
664 B
Python
from cakechat.config import PREDICTION_MODES
|
|
|
|
# Prediction mode used in API
|
|
PREDICTION_MODE = PREDICTION_MODES.sampling_reranking
|
|
|
|
# In case of PREDICTION_MODES.{beamsearch, beamsearch-reranking, sampling-reranking} choose random non-offensive
|
|
# response out of K best candidates proposed by the algorithm.
|
|
NUM_BEST_CANDIDATES_TO_PICK_FROM = 3
|
|
|
|
# In case of PREDICTION_MODES.sampling generate samples one-by-one until a non-offensive sample occurs. This parameter
|
|
# defines max number of samples will be generated until succeed.
|
|
SAMPLING_ATTEMPTS_NUM = 10
|
|
|
|
# Default response text in case we weren't able to produce appropriate response.
|
|
DEFAULT_RESPONSE = '🙊'
|