Views
No views yet
1from transformers import AutoTokenizer, T5ForConditionalGeneration
2tokenizer = AutoTokenizer.from_pretrained("laituan245/molt5-base", model_max_length=512)
3model = T5ForConditionalGeneration.from_pretrained('PhTae/MolBridge-Gen-Base-S2C')
4
5canonicalized_smiles = 'CC(=O)N[C@@H](CCCN=C(N)N)C(=O)[O-]'
6canonicalized_smiles = 'Provide a whole descriptions of this molecule: ' + canonicalized_smiles
7
8token = tokenizer(canonicalized_smiles, return_tensors='pt', padding='longest', truncation=True)
9gen_results = model.generate(input_ids=token['input_ids'],
10 attention_mask=token['attention_mask'],
11 num_beams=5,
12 max_new_tokens=512)
13
14gen_results = tokenizer.decode(gen_results[0], skip_special_tokens=True)
15print(gen_results)