fix for plus lora

ver5
hako-mikan 2023-03-06 12:34:57 +09:00 committed by GitHub
parent 48acc7a632
commit 8310e1e9f2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions

View File

@ -313,7 +313,8 @@ def pluslora(lnames,loraratios,settings,output,model,precision):
).unsqueeze(2).unsqueeze(3) * scale
theta_0[keychanger[msd_key]] = torch.nn.Parameter(weight)
#usemodelgen(theta_0,model)
result = savemodel(theta_0,dname,output,settings.append(precision),model)
settings.append(precision)
result = savemodel(theta_0,dname,output,settings,model)
del theta_0
gc.collect()
return result