zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 17:31:18 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

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

Browse Source 
* 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 37b41cec60
commit 37856f70a8
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lora/lora.py
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

1
lora/requirements.txt
View File

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