ViQA / config.py
CreatorPhan's picture
Duplicate from CreatorPhan/VDT
56523b5
raw
history blame contribute delete
231 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 = 16
show_c = 8