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

Remove unused code part

Browse Source
This commit is contained in:
Shunta Saito 2025-02-14 23:06:42 +09:00
parent 66dd97ed3d
commit 103c6616c4
1 changed files with 0 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1648,9 +1648,6 @@ class Model(PlamoPreTrainedModel):
raise ValueError(
f"Unexpected output type for causal language model: {type(output)} != CausalLMOutputWithPast"
)
if output.past_key_values is not None:
# output.past_key_values is actually a PlamoCache object
self._cache = output.past_key_values # type: ignore
if output.logits is not None:
return output.logits
else: