BMP commited on
Commit
d2cf923
·
verified ·
1 Parent(s): 18c9410

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +5 -1
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(config["input_dim"], config["embedding_dim"])
 
 
 
 
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