Update index.html

index.html

@@ -82,16 +82,16 @@
                 'Authorization': `Bearer ${hf_token}`
             }
         }
-        let configRes = await fetch(
+        let config_res = await fetch(
           `https://huggingface.co/${hf_model}/raw/main/config.json`, auth
         )
-        if (configRes.status === 401) {
+        if (config_res.status === 401) {
           throw new Error("Model is either private or gated, you must provide an access token")
         }
-        if (configRes.status === 403) {
+        if (config_res.status === 403) {
           throw new Error("Model is either private or gated and provided access token does not have access to the repo")
         }
-        let config = await configRes.json()
+        let config = await config_res.json()
         let model_size = 0
         let dtype = parseInt(config["torch_dtype"].replaceAll(/\D/g, '')) / 8
         try {
@@ -106,20 +106,22 @@
               throw new Error("no size in pytorch metadata")
             }
           } catch {
-            throw new Error("Neither model.safetensors.index.json nor pytorch_model.bin.index.json contain size of model")
-          //   huggingface is now blocking corsproxy
-          //   let model_page = await fetch(
-          //       "https://corsproxy.io/?" + encodeURIComponent(`https://huggingface.co/${hf_model}`)
-          //   ).then(r => r.text())
-          //   let el = document.createElement( 'html' );
-          //   el.innerHTML = model_page
-          //   let params_el = el.querySelector('div[data-target="ModelSafetensorsParams"]')
-          //   if (params_el !== null) {
-          //     model_size = JSON.parse(params_el.attributes.getNamedItem("data-props").value)["safetensors"]["total"]
-          //   } else {
-          //     params_el = el.querySelector('div[data-target="ModelHeader"]')
-          //     model_size = JSON.parse(params_el.attributes.getNamedItem("data-props").value)["model"]["safetensors"]["total"]
-          //   }
+            try {
+              let model_page = await fetch(
+                `https://huggingface.co/${hf_model}`
+              ).then(r => r.text())
+              let el = document.createElement( 'html' );
+              el.innerHTML = model_page
+              let params_el = el.querySelector('div[data-target="ModelSafetensorsParams"]')
+              if (params_el !== null) {
+                model_size = JSON.parse(params_el.attributes.getNamedItem("data-props").value)["safetensors"]["total"]
+              } else {
+                params_el = el.querySelector('div[data-target="ModelHeader"]')
+                model_size = JSON.parse(params_el.attributes.getNamedItem("data-props").value)["model"]["safetensors"]["total"]
+              }
+            } catch {
+              throw new Error("Couldn't determine model size from safetensor/pytorch index metadata nor from the model card. If the model is an unsharded pytorch model, it is not supported by this calculator.")
+            }
           }
         }
         config.parameters = model_size