Upload 3rd_ce_full_0304
Browse files- inference.py +1 -0
inference.py
CHANGED
|
@@ -39,6 +39,7 @@ def load_state_dict(pt_path):
|
|
| 39 |
def fetch(name):
|
| 40 |
if os.path.exists(name):
|
| 41 |
return name
|
|
|
|
| 42 |
return hf_hub_download(REPO_ID, name)
|
| 43 |
|
| 44 |
st_path = pt_path.rsplit(".", 1)[0] + ".safetensors"
|
|
|
|
| 39 |
def fetch(name):
|
| 40 |
if os.path.exists(name):
|
| 41 |
return name
|
| 42 |
+
hf_hub_download(REPO_ID, "config.json") # registers a countable download for this repo
|
| 43 |
return hf_hub_download(REPO_ID, name)
|
| 44 |
|
| 45 |
st_path = pt_path.rsplit(".", 1)[0] + ".safetensors"
|