Skip to content

Commit e01c9ab

Browse files
committed
minior fix
minior fix minior fix
1 parent ce20bed commit e01c9ab

File tree

2 files changed

+1
-7
lines changed

2 files changed

+1
-7
lines changed

protenix/openfold_local/model/primitives.py

Lines changed: 0 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -544,13 +544,6 @@ def forward(
544544

545545
if use_memory_efficient_kernel:
546546
raise Exception(f"use_memory_efficient_kernel=True not supported!!!")
547-
if len(biases) > 2:
548-
raise ValueError(
549-
"If use_memory_efficient_kernel is True, you may only "
550-
"provide up to two bias terms"
551-
)
552-
o = attention_core(q, k, v, *((biases + [None] * 2)[:2]))
553-
o = o.transpose(-2, -3)
554547
elif use_deepspeed_evo_attention:
555548
if len(biases) > 2:
556549
raise ValueError(

scripts/msa/step1-get_prot_seq.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -30,6 +30,7 @@ def get_seqs(mmcif_file):
3030

3131
entity_poly = mmcif_parser.get_category_table("entity_poly")
3232
if entity_poly is None:
33+
pdb_id = mmcif_file.name.split(".")[0]
3334
return pdb_id, None
3435
entity_poly["mmcif_seq_old"] = entity_poly.pdbx_seq_one_letter_code_can.str.replace(
3536
"\n", ""

0 commit comments

Comments
 (0)