From 757e6af0af2f8de6da696976f3110cd70085adad Mon Sep 17 00:00:00 2001 From: Volpeon Date: Tue, 7 Feb 2023 20:56:37 +0100 Subject: Fix Lora memory usage --- training/strategy/lora.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) (limited to 'training/strategy/lora.py') 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 -- cgit v1.2.3-54-g00ecf