We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 2981fa2 commit e3f34a6Copy full SHA for e3f34a6
scripts/huggingface/upload_model.py
@@ -20,16 +20,7 @@
20
21
def retrive_model_path(model: torch.nn.Module) -> pathlib.Path:
22
model_folder = inspect.getfile(model.__class__).split("/")[-2]
23
- model_path = (
24
- pathlib.Path(__file__)
25
- / ".."
26
27
28
- / "the_well"
29
- / "benchmark"
30
- / "models"
31
- / model_folder
32
- ).resolve()
+ model_path = (CONFIG_DIR.parent / "models" / model_folder).resolve()
33
return model_path
34
35
0 commit comments