t5vi / create_config.py
imthanhlv's picture
Saving weights and logs of step 10000
94b5353
raw
history blame contribute delete
140 Bytes
from transformers import T5Config
config = T5Config.from_pretrained("google/t5-v1_1-base", vocab_size=32_000)
config.save_pretrained("./")