Update README.md
Browse files
README.md
CHANGED
|
@@ -35,8 +35,8 @@ from flax import nnx, serialization
|
|
| 35 |
from huggingface_hub import hf_hub_download
|
| 36 |
|
| 37 |
# Download model weights and config
|
| 38 |
-
model_path = hf_hub_download(repo_id="
|
| 39 |
-
config_path = hf_hub_download(repo_id="
|
| 40 |
|
| 41 |
# Load config to get model architecture
|
| 42 |
import importlib.util
|
|
|
|
| 35 |
from huggingface_hub import hf_hub_download
|
| 36 |
|
| 37 |
# Download model weights and config
|
| 38 |
+
model_path = hf_hub_download(repo_id="jcopo/mnist", filename="model.msgpack")
|
| 39 |
+
config_path = hf_hub_download(repo_id="jcopo/mnist", filename="config.py")
|
| 40 |
|
| 41 |
# Load config to get model architecture
|
| 42 |
import importlib.util
|