fix bnb lora

This commit is contained in:
layerdiffusion
2024-08-16 21:09:14 -07:00
parent fc0c03b793
commit f3e211d431
2 changed files with 11 additions and 3 deletions

View File

@@ -76,6 +76,15 @@ def get_attr(obj, attr):
return obj
def get_attr_with_parent(obj, attr):
attrs = attr.split(".")
parent = obj
for name in attrs:
parent = obj
obj = getattr(obj, name)
return parent, obj
def calculate_parameters(sd, prefix=""):
params = 0
for k in sd.keys():