(self)
| 632 | |
| 633 | @pytest.fixture |
| 634 | def large_model(self): |
| 635 | class LargeModel(nn.Module): |
| 636 | def __init__(self): |
| 637 | super().__init__() |
| 638 | self.lin0 = nn.Linear(10, 10) |
| 639 | self.conv0 = nn.Conv2d(3, 10, 3) |
| 640 | self.emb0 = nn.Embedding(10, 10) |
| 641 | self.lin1 = nn.Linear(10, 10) |
| 642 | self.conv1 = nn.Conv2d(3, 10, 3) |
| 643 | self.emb1 = nn.Embedding(10, 10) |
| 644 | |
| 645 | config0 = LoraConfig(target_modules=["lin0", "conv1", "emb0"]) |
| 646 | config1 = LoraConfig(target_modules=["lin0", "lin1"], r=16) |
| 647 | model = get_peft_model(LargeModel(), config0) |
| 648 | model.add_adapter("other", config1) |
| 649 | return model |
| 650 | |
| 651 | ################ |
| 652 | # layer status # |
nothing calls this directly
no test coverage detected