Skip to content

Commit

Permalink
remove parallel grad checkpointing test
Browse files Browse the repository at this point in the history
  • Loading branch information
calpt committed Nov 25, 2024
1 parent 8421f63 commit 5edfd4d
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions tests/methods/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -422,16 +422,3 @@ def adapter_setup_fn(model):
model.adapter_to("adapter1", torch_device)

self._run_gradient_checkpointing_test_helper(adapter_setup_fn)

def run_gradient_checkpointing_test_parallel_adapters(self, adapter_config):
def adapter_setup_fn(model):
model.add_adapter("adapter1", config=adapter_config)
model.add_adapter("adapter2", config=adapter_config)
self.add_head(model, "adapter1")
self.add_head(model, "adapter2")
model.active_adapters = ac.Parallel("adapter1", "adapter2")
model.train_adapter(ac.Parallel("adapter1", "adapter2"))
model.adapter_to("adapter1", torch_device)
model.adapter_to("adapter2", torch_device)

self._run_gradient_checkpointing_test_helper(adapter_setup_fn)

0 comments on commit 5edfd4d

Please sign in to comment.