Views
No views yet
T5) architecture:Hugging Face Accelerate1from transformers import T5ForConditionalGeneration, PreTrainedTokenizerFast
2
3def postprocess_rna(rna):
4 return rna.replace('b', 'A').replace('j', 'C').replace(
5 'u', 'U').replace('z', 'G').replace(' ', '').replace(
6 'B', 'A').replace('J', 'C').replace('U', 'U').replace('Z', 'G')
7
8# Load model
9model = T5ForConditionalGeneration.from_pretrained("SobhanShukueian/rnatranslator")
10
11# Load separate tokenizers
12protein_tokenizer = PreTrainedTokenizerFast.from_pretrained("SobhanShukueian/rnatranslator", subfolder="protein_tokenizer")
13rna_tokenizer = PreTrainedTokenizerFast.from_pretrained("SobhanShukueian/rnatranslator", subfolder="rna_tokenizer")
14
15
16protein_seq = "MSGGGVIRGPAGNNDCRIYVGNLPPDIRTKDIEDVFYKYGAIRDIDLKNRRGGPPFAFVEFEDPRDAEDAVYGRDGYDYDGYRLRVEFPRSGRGTGRGGGGGGGGGAPRGRYGPPSRRSENRVVVSGLPPSGSWQDLKDHMREAGDVCYADVYRDGTGVVEFVRKEDMTYAVRKLDNTKFRSHEGETAYIRVKVDGPRSPSYGRSRSRSRSRSRSRSRSNSRSRSYSPRRSRGSPRYSPRHSRSRSRT"
17inputs = protein_tokenizer(protein_seq, return_tensors="pt").input_ids
18
19# Generate RNA
20gen_args = {
21 'max_length': 256,
22 'repetition_penalty': 1.5,
23 'encoder_repetition_penalty': 1.3,
24 'num_return_sequences': 1,
25 'top_k': 30,
26 'temperature': 1.5,
27 'num_beams': 1,
28 'do_sample': True,
29}
30
31outputs = model.generate(inputs, **gen_args)
32rna_sequence = rna_tokenizer.decode(outputs[0], skip_special_tokens=True)
33print(postprocess_rna(rna_sequence))1- model.safetensors – Fine-tuned model weights
2- config.json – Model configuration
3- protein_tokenizer/ – Tokenizer for protein sequences (encoder)
4- rna_tokenizer/ – Tokenizer for RNA sequences (decoder)
5- README.md – This model card