Skip to content

Commit

Permalink
Remove outdated assert for layer checkpointing.
Browse files Browse the repository at this point in the history
Looks like this assert checks abandoned experimental `layer_drop` param, which not exists anymore.
  • Loading branch information
escfrya authored Nov 10, 2023
1 parent 5905d2e commit ab95eeb
Showing 1 changed file with 0 additions and 1 deletion.
1 change: 0 additions & 1 deletion audiocraft/modules/transformer.py
Original file line number Diff line number Diff line change
Expand Up @@ -648,7 +648,6 @@ def __init__(self, d_model: int, num_heads: int, num_layers: int, dim_feedforwar
# see audiocraft/optim/fsdp.py, magic signal to indicate this requires fixing the
# backward hook inside of FSDP...
layer._magma_checkpointed = True # type: ignore
assert layer.layer_drop == 0., "Need further checking" # type: ignore

def _apply_layer(self, layer, *args, **kwargs):
method = self.checkpointing
Expand Down

0 comments on commit ab95eeb

Please sign in to comment.