Adjust download path for weights file
Browse files
nemotron_page_elements_v3/model.py
CHANGED
|
@@ -15,7 +15,7 @@ from nemotron_page_elements_v3.yolox.boxes import postprocess
|
|
| 15 |
|
| 16 |
# HuggingFace repository for downloading model weights
|
| 17 |
HF_REPO_ID = "nvidia/nemotron-page-elements-v3"
|
| 18 |
-
WEIGHTS_FILENAME = "weights.pth"
|
| 19 |
|
| 20 |
|
| 21 |
def define_model(config_name: str = "page_element_v3", verbose: bool = True) -> nn.Module:
|
|
|
|
| 15 |
|
| 16 |
# HuggingFace repository for downloading model weights
|
| 17 |
HF_REPO_ID = "nvidia/nemotron-page-elements-v3"
|
| 18 |
+
WEIGHTS_FILENAME = "nemotron_page_elements_v3/weights.pth"
|
| 19 |
|
| 20 |
|
| 21 |
def define_model(config_name: str = "page_element_v3", verbose: bool = True) -> nn.Module:
|