zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-11-02 09:18:11 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Get trellis to run

Browse Source
This commit is contained in:
Awni Hannun
2025-04-26 07:02:20 -07:00
parent e3d275bc49
commit 998404ada4
4 changed files with 33 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
python/mlx/nn/layers/linear.py
View File

@@ -6,7 +6,6 @@ from typing import Any, Literal
import mlx.core as mx
from mlx.nn.layers.base import Module
from mlx.nn.layers.quantized import QuantizedLinear
from mlx.nn.layers.viterbi import quantize as trellis_quantize
class Identity(Module):

1
python/mlx/nn/layers/quantized.py
View File

@@ -5,7 +5,6 @@ from typing import Callable, Literal, Optional, Union
import mlx.core as mx
from mlx.nn.layers.base import Module
from mlx.nn.layers.viterbi import quantize as trellis_quantize
from mlx.utils import tree_map_with_path