zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-06-24 09:21: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 math
import time import time
from pathlib import Path from pathlib import Path
from typing import List, Optional, Tuple from typing import List
import mlx.core as mx import mlx.core as mx
import mlx.nn as nn import mlx.nn as nn
import mlx.optimizers as optim import mlx.optimizers as optim
import numpy as np 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 models import LoRALinear, Model, ModelArgs
from sentencepiece import SentencePieceProcessor from sentencepiece import SentencePieceProcessor

1
lora/requirements.txt
View File

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