Skip to content

GitLab

"...git@developer.sourcefind.cn:OpenDAS/openpcdet.git" did not exist on "4d8624fd785ba7197d56355db16d6fd262cce402"
Commit 1271a03f authored by Christina Floristean's avatar Christina Floristean
Browse files

Use ds kernel for default long seq inference

parent 71526dff
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
openfold/config.py
View file @ 1271a03f
...@@ -200,9 +200,9 @@ def model_config( ...@@ -200,9 +200,9 @@ def model_config(
if long_sequence_inference: if long_sequence_inference:
assert(not train) assert(not train)
c.globals.offload_inference = True c.globals.offload_inference = True
c.globals.use_lma = True # Default to DeepSpeed memory-efficient attention kernel unless use_lma is explicitly set
c.globals.use_deepspeed_evo_attention = True if not c.globals.use_lma else False
c.globals.use_flash = False c.globals.use_flash = False
c.globals.use_deepspeed_evo_attention = False
c.model.template.offload_inference = True c.model.template.offload_inference = True
c.model.template.template_pair_stack.tune_chunk_size = False c.model.template.template_pair_stack.tune_chunk_size = False
c.model.extra_msa.extra_msa_stack.tune_chunk_size = False c.model.extra_msa.extra_msa_stack.tune_chunk_size = False
......
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