```py | |
checkpoint = "google/t5-v1_1-xxl" | |
model = T5EncoderModel.from_pretrained( | |
checkpoint, torch_dtype=torch.bfloat16 | |
) | |
tokenizer = T5Tokenizer.from_pretrained( | |
checkpoint | |
) | |
model.push_to_hub("ariG23498/t5-v1_1-xxl-torch") | |
tokenizer.push_to_hub("ariG23498/t5-v1_1-xxl-torch")``` |