Helw150
commited on
Commit
•
b5e9bbc
1
Parent(s):
fcf1cf7
Need a Config DL
Browse files- modeling_diva.py +6 -0
modeling_diva.py
CHANGED
|
@@ -130,6 +130,12 @@ class DiVAModel(PreTrainedModel):
|
|
| 130 |
token=kwargs.get("token", None),
|
| 131 |
local_dir=os.path.dirname(__file__),
|
| 132 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 133 |
via_path = os.path.dirname(__file__) + "/model-00001-of-00004.safetensors"
|
| 134 |
config_path = os.path.dirname(__file__) + "/config.json"
|
| 135 |
with open(config_path, "r") as f:
|
|
|
|
| 130 |
token=kwargs.get("token", None),
|
| 131 |
local_dir=os.path.dirname(__file__),
|
| 132 |
)
|
| 133 |
+
hf_hub_download(
|
| 134 |
+
repo_id=pretrained_model_name_or_path,
|
| 135 |
+
filename="config.json",
|
| 136 |
+
token=kwargs.get("token", None),
|
| 137 |
+
local_dir=os.path.dirname(__file__),
|
| 138 |
+
)
|
| 139 |
via_path = os.path.dirname(__file__) + "/model-00001-of-00004.safetensors"
|
| 140 |
config_path = os.path.dirname(__file__) + "/config.json"
|
| 141 |
with open(config_path, "r") as f:
|