from local

This commit is contained in:
dohe0342 2023-05-25 21:05:37 +09:00
parent 5c3819098d
commit c46a29212f
2 changed files with 2 additions and 2 deletions

View File

@ -139,8 +139,8 @@ class LoRAHook():
) )
def hook_fn(self, module, input, output): def hook_fn(self, module, input, output):
#print(input) #print(input)
print(module, input, output[0].size()) #print(module, input, output[0].size())
#lora_out = self.lora(input) lora_out = self.lora(input)
#output += lora_out #output += lora_out
def save_checkpoint(self, i, iter_, save_dir): def save_checkpoint(self, i, iter_, save_dir):