Update index.html
Browse files- index.html +22 -16
index.html
CHANGED
|
@@ -82,34 +82,40 @@
|
|
| 82 |
'Authorization': `Bearer ${hf_token}`
|
| 83 |
}
|
| 84 |
}
|
| 85 |
-
let
|
| 86 |
`https://huggingface.co/${hf_model}/raw/main/config.json`, auth
|
| 87 |
-
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 88 |
let model_size = 0
|
| 89 |
try {
|
| 90 |
model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/model.safetensors.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / 2
|
| 91 |
if (isNaN(model_size)) {
|
| 92 |
-
throw new
|
| 93 |
}
|
| 94 |
} catch (e) {
|
| 95 |
try {
|
| 96 |
model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/pytorch_model.bin.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / 2
|
| 97 |
if (isNaN(model_size)) {
|
| 98 |
-
throw new
|
| 99 |
}
|
| 100 |
} catch {
|
| 101 |
-
|
| 102 |
-
|
| 103 |
-
|
| 104 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 108 |
-
|
| 109 |
-
|
| 110 |
-
|
| 111 |
-
|
| 112 |
-
|
|
|
|
|
|
|
| 113 |
}
|
| 114 |
}
|
| 115 |
config.parameters = model_size
|
|
|
|
| 82 |
'Authorization': `Bearer ${hf_token}`
|
| 83 |
}
|
| 84 |
}
|
| 85 |
+
let configRes = await fetch(
|
| 86 |
`https://huggingface.co/${hf_model}/raw/main/config.json`, auth
|
| 87 |
+
)
|
| 88 |
+
if (configRes.status === 403) {
|
| 89 |
+
throw new Error("Model is either private or gated and you have not provided an access token or the provided access token does not have access to the repo")
|
| 90 |
+
}
|
| 91 |
+
let config = await configRes.json()
|
| 92 |
let model_size = 0
|
| 93 |
try {
|
| 94 |
model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/model.safetensors.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / 2
|
| 95 |
if (isNaN(model_size)) {
|
| 96 |
+
throw new Error("no size in safetensors metadata")
|
| 97 |
}
|
| 98 |
} catch (e) {
|
| 99 |
try {
|
| 100 |
model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/pytorch_model.bin.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / 2
|
| 101 |
if (isNaN(model_size)) {
|
| 102 |
+
throw new Error("no size in pytorch metadata")
|
| 103 |
}
|
| 104 |
} catch {
|
| 105 |
+
throw new Error("Neither model.safetensors.index.json nor pytorch_model.bin.index.json contain size of model")
|
| 106 |
+
// huggingface is now blocking corsproxy
|
| 107 |
+
// let model_page = await fetch(
|
| 108 |
+
// "https://corsproxy.io/?" + encodeURIComponent(`https://huggingface.co/${hf_model}`)
|
| 109 |
+
// ).then(r => r.text())
|
| 110 |
+
// let el = document.createElement( 'html' );
|
| 111 |
+
// el.innerHTML = model_page
|
| 112 |
+
// let params_el = el.querySelector('div[data-target="ModelSafetensorsParams"]')
|
| 113 |
+
// if (params_el !== null) {
|
| 114 |
+
// model_size = JSON.parse(params_el.attributes.getNamedItem("data-props").value)["safetensors"]["total"]
|
| 115 |
+
// } else {
|
| 116 |
+
// params_el = el.querySelector('div[data-target="ModelHeader"]')
|
| 117 |
+
// model_size = JSON.parse(params_el.attributes.getNamedItem("data-props").value)["model"]["safetensors"]["total"]
|
| 118 |
+
// }
|
| 119 |
}
|
| 120 |
}
|
| 121 |
config.parameters = model_size
|