File test_t5.py

SwissArmyTransformer/examples/t5/test_t5.py:None–None · view source on GitHub ↗

Source from the content-addressed store, hash-verified

1	from transformers import T5Model, T5ForConditionalGeneration, T5Tokenizer
2	device = 'cuda:1'
3	tokenizer = T5Tokenizer.from_pretrained("t5-large")
4	model = T5ForConditionalGeneration.from_pretrained("/dataset/fd5061f6/yanan/huggingface_models/t5-xl-lm-adapt")

nothing calls this directly

toMethod · 0.80

from_pretrainedMethod · 0.45

backwardMethod · 0.45

no test coverage detected