from local

This commit is contained in:
dohe0342 2023-05-24 13:46:28 +09:00
parent 32fac34586
commit a53453c22c
3 changed files with 0 additions and 1 deletions

View File

@ -142,7 +142,6 @@ 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: