Skip to content

Commit b7812cc

Browse files
authored
Update d_smiles2pv.py
1 parent 4fd4236 commit b7812cc

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

d_smiles2pv.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def pv_generate(model, data_loader):
3838
# convert list of string to dataloader
3939
if isinstance(data_loader, list):
4040
if data_loader[0][5] != "[CLS]":
41-
data_loader = ['[CLS]'+d for d in data_loaer]
41+
data_loader = ['[CLS]'+d for d in data_loader]
4242
gather = []
4343
text_input = tokenizer(data_loader, padding='longest', truncation=True, max_length=100, return_tensors="pt").to(device)
4444
text_embeds = model.text_encoder.bert(text_input.input_ids[:, 1:], attention_mask=text_input.attention_mask[:, 1:],

0 commit comments

Comments
 (0)