zhangyiss/mlx-examples
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx-examples.git synced 2025-08-29 12:26:07 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Apply formatter

Browse Source
This commit is contained in:
Shunta Saito 2025-02-13 20:02:38 +09:00
parent 07cf4336b3
commit fb5e225523
1 changed files with 3 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
llms/mlx_lm/models/plamo2.py
View File

@ -1947,12 +1947,7 @@ class Model(PlamoPreTrainedModel):
weights[k] = v.moveaxis(2, 1) weights[k] = v.moveaxis(2, 1)
return weights return weights
def make_cache(self) -> PlamoCache:
print("make_cache")
return "a"
def __call__(self, inputs: mx.array, cache: PlamoCache | None = None) -> mx.array: def __call__(self, inputs: mx.array, cache: PlamoCache | None = None) -> mx.array:
print(cache)
output = self.forward( output = self.forward(
input_ids=inputs, input_ids=inputs,
use_cache=self.config.use_cache, use_cache=self.config.use_cache,