We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 38cbf51 commit d1c15feCopy full SHA for d1c15fe
mlx-swift-lm
@@ -1 +1 @@
1
-Subproject commit c923413c1ce39e6c58e1ce8ac898ffda65c3b56e
+Subproject commit 689b6ae60d3451331b079f45d897431db01c1dec
test_keys.swift
@@ -1,9 +1,7 @@
import Foundation
2
import MLX
3
+import MLXNN
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
-}
+MLX.GPU.set(cacheLimit: 10 * 1024 * 1024)
+let norm = RMSNorm(dimensions: 128)
+print("RMSNorm parameters:", norm.parameters().keys)
0 commit comments