VDT / config.py
CreatorPhan's picture
change to base
0e4456d
raw
history blame
215 Bytes
class Config:
# model = 'flan-small'
# tokenizer = 'CreatorFPT/T5-base'
model = 'vietAI-base-MRC'
tokenizer = 'VietAI/vit5-base'
seq_len = 512
out_len = 170
device = 'cpu'
choices = 8