1 files changed, 1 insertions, 3 deletions
diff --git a/training/strategy/lora.py b/training/strategy/lora.py
index 88d1824..92abaa6 100644
--- a/training/strategy/lora.py
+++ b/training/strategy/lora.py
@@ -1,11 +1,9 @@
-from contextlib import nullcontext
 from typing import Optional
 from functools import partial
-from contextlib import contextmanager, nullcontext
+from contextlib import contextmanager
 from pathlib import Path
 import torch
-import torch.nn as nn
 from torch.utils.data import DataLoader
 from accelerate import Accelerator

diff --git a/training/strategy/lora.py b/training/strategy/lora.py index 88d1824..92abaa6 100644 --- a/training/strategy/lora.py +++ b/training/strategy/lora.py
@@ -1,11 +1,9 @@
1	from contextlib import nullcontext
2	from typing import Optional	1	from typing import Optional
3	from functools import partial	2	from functools import partial
4	from contextlib import contextmanager, nullcontext	3	from contextlib import contextmanager
5	from pathlib import Path	4	from pathlib import Path
6		5
7	import torch	6	import torch
8	import torch.nn as nn
9	from torch.utils.data import DataLoader	7	from torch.utils.data import DataLoader
10		8
11	from accelerate import Accelerator	9	from accelerate import Accelerator