Skip to content

GitLab

"configs/git@developer.sourcefind.cn:OpenDAS/mmgeneration.git" did not exist on "57e0e89170c8aa97b0980fafaf87b2df0204f93d"
Commit 91f031f5 authored by Gustaf Ahdritz's avatar Gustaf Ahdritz
Browse files

Fix formatting

parent b1780f4e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/utils/import_weights.py
View file @ 91f031f5
...@@ -397,8 +397,7 @@ def generate_translation_dict(model, version): ...@@ -397,8 +397,7 @@ def generate_translation_dict(model, version):
tps_blocks = model.template_pair_stack.blocks tps_blocks = model.template_pair_stack.blocks
tps_blocks_params = stacked( tps_blocks_params = stacked(
[TemplatePairBlockParams(b) for b in tps_blocks] [TemplatePairBlockParams(b) for b in tps_blocks]
) )
template_param_dict = { template_param_dict = {
"template_embedding": { "template_embedding": {
"single_template_embedding": { "single_template_embedding": {
...@@ -421,6 +420,7 @@ def generate_translation_dict(model, version): ...@@ -421,6 +420,7 @@ def generate_translation_dict(model, version):
model.template_angle_embedder.linear_2 model.template_angle_embedder.linear_2
), ),
} }
translations["evoformer"].update(template_param_dict) translations["evoformer"].update(template_param_dict)
if "_ptm" in version: if "_ptm" in version:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment