zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-09-10 09:45: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 6b71e18a0a
commit 91017ac223
2 changed files with 3 additions and 2 deletions
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