Update index.html

index.html

@@ -93,14 +93,15 @@
         }
         let config = await configRes.json()
         let model_size = 0
+        let dtype = parseInt(config["torch_dtype"].replaceAll(/\D/g, '')) / 8
         try {
-          model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/model.safetensors.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / 2
+          model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/model.safetensors.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / dtype
           if (isNaN(model_size)) {
             throw new Error("no size in safetensors metadata")
           }
         } catch (e) {
           try {
-            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
+            model_size = (await fetch(`https://huggingface.co/${hf_model}/resolve/main/pytorch_model.bin.index.json`, auth).then(r => r.json()))["metadata"]["total_size"] / dtype
             if (isNaN(model_size)) {
               throw new Error("no size in pytorch metadata")
             }