Spaces:

webai-community
/

demos

Running

Yang Gu commited on Apr 29, 2024

Commit

27da4cd

1 Parent(s): 3b919a3

Adjust model path

Files changed (2) hide show

demo/ort-phi2/index.html CHANGED Viewed

@@ -6,7 +6,8 @@
 </head>
 <body>
-  <script src="https://cdn.jsdelivr.net/npm/onnxruntime-web@dev/dist/ort.webgpu.min.js"> </script>
   <!-- <script src="https://wp-27.sh.intel.com/workspace/project/onnxruntime/js/web/dist/ort.webgpu.min.js"> </script> -->
   <script type="module">
@@ -79,15 +80,15 @@
         this.profiler = options.profiler;
         this.trace = options.trace;
-        const model_path = (local) ? "https://wp-27.sh.intel.com/workspace/project/models/" + model.path : "https://huggingface.co/" + model.path + "/resolve/main";
         log(`loading... ${model.name},  ${provider}`);
-        const json_bytes = await fetchAndCache(model_path + "/config.json");
         let textDecoder = new TextDecoder();
         const model_config = JSON.parse(textDecoder.decode(json_bytes));
-        const model_bytes = await fetchAndCache(model_path + "/phi2-int4.onnx");
-        const externaldata = (model.externaldata) ? await fetchAndCache(model_path + '/onnx/decoder_model_merged.onnx.data') : false;
         let modelSize = model_bytes.byteLength;
         if (externaldata) {
           modelSize += externaldata.byteLength;

 </head>
 <body>
+  <script src="../../util.js"></script>
+  <script src="https://cdn.jsdelivr.net/npm/onnxruntime-web@dev/dist/ort.webgpu.min.js"></script>
   <!-- <script src="https://wp-27.sh.intel.com/workspace/project/onnxruntime/js/web/dist/ort.webgpu.min.js"> </script> -->
   <script type="module">
         this.profiler = options.profiler;
         this.trace = options.trace;
+        const modelPath = getModelsPath() + model.path;
         log(`loading... ${model.name},  ${provider}`);
+        const json_bytes = await fetchAndCache(modelPath + "/config.json");
         let textDecoder = new TextDecoder();
         const model_config = JSON.parse(textDecoder.decode(json_bytes));
+        const model_bytes = await fetchAndCache(modelPath + "/phi2-int4.onnx");
+        const externaldata = (model.externaldata) ? await fetchAndCache(modelPath + '/phi2-int4.onnx.data') : false;
         let modelSize = model_bytes.byteLength;
         if (externaldata) {
           modelSize += externaldata.byteLength;

util.js CHANGED Viewed

@@ -74,3 +74,11 @@ async function readResponse(name, response, progressCallback) {
   await read();
   return buffer;
 }

   await read();
   return buffer;
 }
+function getModelsPath() {
+  if (window.location.href.includes('intel')) {
+    return "https://wp-27.sh.intel.com/workspace/project/models/";
+  } else {
+    return "https://huggingface.co/webai-community/models/resolve/main/";
+  }
+}