from local

This commit is contained in:
dohe0342 2023-05-24 13:42:58 +09:00
parent 748908b1d1
commit ecb2f5fdc1
2 changed files with 1 additions and 0 deletions

View File

@ -142,6 +142,7 @@ class LoRAHook():
print(input) print(input)
lora_out = self.lora(input) lora_out = self.lora(input)
output = input + lora_out output = input + lora_out
exit()
def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: def set_batch_count(model: Union[nn.Module, DDP], batch_count: float) -> None: