Upload README.md with huggingface_hub
Browse files
README.md
CHANGED
|
@@ -28,7 +28,11 @@ import json
|
|
| 28 |
with open(f"{model_path}/config.json") as f:
|
| 29 |
config = json.load(f)
|
| 30 |
|
| 31 |
-
model = CAMPPModel(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
weights = mx.load(f"{model_path}/weights.npz")
|
| 33 |
model.load_weights(weights)
|
| 34 |
|
|
|
|
| 28 |
with open(f"{model_path}/config.json") as f:
|
| 29 |
config = json.load(f)
|
| 30 |
|
| 31 |
+
model = CAMPPModel(
|
| 32 |
+
input_dim=config["input_dim"],
|
| 33 |
+
embedding_dim=config["embedding_dim"],
|
| 34 |
+
input_channels=config.get("input_channels", 64)
|
| 35 |
+
)
|
| 36 |
weights = mx.load(f"{model_path}/weights.npz")
|
| 37 |
model.load_weights(weights)
|
| 38 |
|