[Fix] Override __getattr__

pull/1687/head
fanqiNO1 2023-07-13 18:50:37 +08:00
parent dd9403b14c
commit c5525a3049
1 changed files with 7 additions and 5 deletions

View File

@ -1,7 +1,7 @@
# Copyright (c) OpenMMLab. All rights reserved.
import math
import re
from typing import List
from typing import Any, List
import torch
from mmengine.logging import print_log
@ -195,9 +195,11 @@ class LoRAModel(BaseModule):
self.register_load_state_dict_post_hook(_load_state_dict_post_hook)
def get_layer_depth(self, param_name: str, prefix: str = ''):
"""Get the layer-wise depth of a parameter for learning rate decay."""
return self.module.get_layer_depth(param_name, prefix)
def forward(self, *args, **kwargs):
return self.module(*args, **kwargs)
def __getattr__(self, name: str) -> Any:
try:
return super(LoRAModel, self).__getattr__(name)
except AttributeError:
return self.module.__getattribute__(name)