add numpy as a requirement to run lora.py (#238)

* add numpy as a requirement to run lora.py

* removed unused imports
This commit is contained in:
Lawrence Wu
2024-01-05 16:16:28 -08:00
committed by GitHub
parent 6b71e18a0a
commit 91017ac223
2 changed files with 3 additions and 2 deletions

View File

@@ -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

View File

@@ -1,3 +1,4 @@
mlx>=0.0.7
sentencepiece
torch
numpy