From aa79e8ec3c003112a7dbb48bde994a2a8325fd7c Mon Sep 17 00:00:00 2001 From: Lawrence Wu Date: Fri, 5 Jan 2024 15:59:33 -0800 Subject: [PATCH] removed unused imports --- lora/lora.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/lora/lora.py b/lora/lora.py index 8a35a6d4..3eed73fb 100644 --- a/lora/lora.py +++ b/lora/lora.py @@ -5,13 +5,13 @@ import json import math import time from pathlib import Path -from typing import List, Optional, Tuple +from typing import List import mlx.core as mx import mlx.nn as nn import mlx.optimizers as optim import numpy as np -from mlx.utils import tree_flatten, tree_map, tree_unflatten +from mlx.utils import tree_flatten, tree_unflatten from models import LoRALinear, Model, ModelArgs from sentencepiece import SentencePieceProcessor