| import torch | |
| import networks | |
| from modules import patches | |
| class LoraPatches: | |
| def __init__(self): | |
| self.Linear_forward = patches.patch(__name__, torch.nn.Linear, 'forward', networks.network_Linear_forward) | |
| self.Linear_load_state_dict = patches.patch(__name__, torch.nn.Linear, '_load_from_state_dict', networks.network_Linear_load_state_dict) | |
| self.Conv2d_forward = patches.patch(__name__, torch.nn.Conv2d, 'forward', networks.network_Conv2d_forward) | |
| self.Conv2d_load_state_dict = patches.patch(__name__, torch.nn.Conv2d, '_load_from_state_dict', networks.network_Conv2d_load_state_dict) | |
| self.GroupNorm_forward = patches.patch(__name__, torch.nn.GroupNorm, 'forward', networks.network_GroupNorm_forward) | |
| self.GroupNorm_load_state_dict = patches.patch(__name__, torch.nn.GroupNorm, '_load_from_state_dict', networks.network_GroupNorm_load_state_dict) | |
| self.LayerNorm_forward = patches.patch(__name__, torch.nn.LayerNorm, 'forward', networks.network_LayerNorm_forward) | |
| self.LayerNorm_load_state_dict = patches.patch(__name__, torch.nn.LayerNorm, '_load_from_state_dict', networks.network_LayerNorm_load_state_dict) | |
| self.MultiheadAttention_forward = patches.patch(__name__, torch.nn.MultiheadAttention, 'forward', networks.network_MultiheadAttention_forward) | |
| self.MultiheadAttention_load_state_dict = patches.patch(__name__, torch.nn.MultiheadAttention, '_load_from_state_dict', networks.network_MultiheadAttention_load_state_dict) | |
| def undo(self): | |
| self.Linear_forward = patches.undo(__name__, torch.nn.Linear, 'forward') | |
| self.Linear_load_state_dict = patches.undo(__name__, torch.nn.Linear, '_load_from_state_dict') | |
| self.Conv2d_forward = patches.undo(__name__, torch.nn.Conv2d, 'forward') | |
| self.Conv2d_load_state_dict = patches.undo(__name__, torch.nn.Conv2d, '_load_from_state_dict') | |
| self.GroupNorm_forward = patches.undo(__name__, torch.nn.GroupNorm, 'forward') | |
| self.GroupNorm_load_state_dict = patches.undo(__name__, torch.nn.GroupNorm, '_load_from_state_dict') | |
| self.LayerNorm_forward = patches.undo(__name__, torch.nn.LayerNorm, 'forward') | |
| self.LayerNorm_load_state_dict = patches.undo(__name__, torch.nn.LayerNorm, '_load_from_state_dict') | |
| self.MultiheadAttention_forward = patches.undo(__name__, torch.nn.MultiheadAttention, 'forward') | |
| self.MultiheadAttention_load_state_dict = patches.undo(__name__, torch.nn.MultiheadAttention, '_load_from_state_dict') | |