j.gilyazev
add personalized-chat-bot
0766044
raw history blame
No virus
366 Bytes
{
"PERSONACHAT_DATASET_NAME": "bavard/personachat_truecased",
"MODEL_NAME": "bigscience/bloom-petals",
"INITIAL_PEERS": [],
"NUM_PREFIX_TOKENS": 16,
"DEVICE": "cpu",
"BATCH_SIZE": 4,
"LR": 0.01,
"WEIGHT_DECAY": 0.0,
"NUM_SAMPLES": 1000,
"SEED": 42,
"MODEL_MAX_LENGTH": 256,
"TUNING_MODE": "ptune",
"N_EPOCH": 10,
"PADDING_SIDE": "right"
}