Skip to content

Commit d1c15fe

Browse files
author
Antigravity
committed
chore: update mlx-swift-lm submodule to include native STFT masks and architecture mapping
1 parent 38cbf51 commit d1c15fe

File tree

2 files changed

+5
-7
lines changed

2 files changed

+5
-7
lines changed

mlx-swift-lm

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
Subproject commit c923413c1ce39e6c58e1ce8ac898ffda65c3b56e
1+
Subproject commit 689b6ae60d3451331b079f45d897431db01c1dec

test_keys.swift

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,7 @@
11
import Foundation
22
import MLX
3+
import MLXNN
34

4-
let weights = try! MLX.load(url: URL(fileURLWithPath: "/Users/simba/.cache/huggingface/hub/models--mlx-community--gemma-4-e4b-it-8bit/snapshots/18f3418f2da5426ec6e4967b4c96bdd2d0002ee4/model-00001-of-00004.safetensors"))
5-
for key in weights.keys {
6-
if key.contains("per_layer") {
7-
print(key)
8-
}
9-
}
5+
MLX.GPU.set(cacheLimit: 10 * 1024 * 1024)
6+
let norm = RMSNorm(dimensions: 128)
7+
print("RMSNorm parameters:", norm.parameters().keys)

0 commit comments

Comments
 (0)