Upload usage_example.py with huggingface_hub
Browse files- usage_example.py +2 -1
usage_example.py
CHANGED
|
@@ -13,7 +13,8 @@ def load_model(model_path="."):
|
|
| 13 |
# Initialize model
|
| 14 |
model = CAMPPModel(
|
| 15 |
input_dim=config["input_dim"],
|
| 16 |
-
embedding_dim=config["embedding_dim"]
|
|
|
|
| 17 |
)
|
| 18 |
|
| 19 |
# Load weights
|
|
|
|
| 13 |
# Initialize model
|
| 14 |
model = CAMPPModel(
|
| 15 |
input_dim=config["input_dim"],
|
| 16 |
+
embedding_dim=config["embedding_dim"],
|
| 17 |
+
input_channels=config.get("input_channels", 64)
|
| 18 |
)
|
| 19 |
|
| 20 |
# Load weights
|