Unverified Commit 3a3717cc authored by Gustaf Ahdritz's avatar Gustaf Ahdritz Committed by GitHub
Browse files

Adjust formatting in evoformer.py

parent 484ce867
...@@ -551,11 +551,9 @@ class ExtraMSAStack(nn.Module): ...@@ -551,11 +551,9 @@ class ExtraMSAStack(nn.Module):
**kwargs, **kwargs,
): ):
super(ExtraMSAStack, self).__init__() super(ExtraMSAStack, self).__init__()
self.clear_cache_between_blocks = clear_cache_between_blocks self.clear_cache_between_blocks = clear_cache_between_blocks
self.blocks = nn.ModuleList() self.blocks = nn.ModuleList()
for _ in range(no_blocks): for _ in range(no_blocks):
block = ExtraMSABlock( block = ExtraMSABlock(
c_m=c_m, c_m=c_m,
......
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