We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 4fd4236 commit b7812ccCopy full SHA for b7812cc
d_smiles2pv.py
@@ -38,7 +38,7 @@ def pv_generate(model, data_loader):
38
# convert list of string to dataloader
39
if isinstance(data_loader, list):
40
if data_loader[0][5] != "[CLS]":
41
- data_loader = ['[CLS]'+d for d in data_loaer]
+ data_loader = ['[CLS]'+d for d in data_loader]
42
gather = []
43
text_input = tokenizer(data_loader, padding='longest', truncation=True, max_length=100, return_tensors="pt").to(device)
44
text_embeds = model.text_encoder.bert(text_input.input_ids[:, 1:], attention_mask=text_input.attention_mask[:, 1:],
0 commit comments