mirror of
https://github.com/wassname/peft.git
synced 2026-09-09 11:28:32 +08:00
Apply suggestions from code review
Co-authored-by: Sourab Mangrulkar <13534540+pacman100@users.noreply.github.com>
This commit is contained in:
co-authored by
Sourab Mangrulkar
parent
ad69958e52
commit
16182ea972
@@ -255,9 +255,6 @@ class LoraModel(PushToHubMixin, torch.nn.Module):
|
||||
return model
|
||||
|
||||
|
||||
def forward(self, *args, **kwargs):
|
||||
return self.model(*args, **kwargs)
|
||||
|
||||
def __getattr__(self, name: str):
|
||||
"""Forward missing attributes to the wrapped module."""
|
||||
try:
|
||||
|
||||
@@ -124,6 +124,7 @@ class PeftConfigMixin(PushToHubMixin):
|
||||
return json_object
|
||||
|
||||
|
||||
@dataclass
|
||||
class PeftConfig(PeftConfigMixin):
|
||||
"""
|
||||
This is the base configuration class to store the configuration of a :class:`~peft.PeftModel`.
|
||||
|
||||
Reference in New Issue
Block a user