Function setup_model

distributed/FSDP/utils/train_utils.py:99–102

(model_name)

def setup_model(model_name):
        model = T5ForConditionalGeneration.from_pretrained(model_name)
        tokenizer =  T5Tokenizer.from_pretrained(model_name, legacy=False)
        return model, tokenizer