Views
No views yet
1import torch
2import numpy as np
3from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, Seq2SeqTrainer, Seq2SeqTrainingArguments
4from datasets import load_dataset
5
6tokenizer = AutoTokenizer.from_pretrained("dnagpt/t5-gene-eng-small")
7print(tokenizer.tokenize("ATCGGAAGGTAAGGGAAAGCG"))
8
9from transformers import AutoModelForSeq2SeqLM
10model = AutoModelForSeq2SeqLM.from_pretrained("dnagpt/t5-gene-eng-small", from_flax=True)
11print(model.config)
12