Yang Gu commited on
Commit
d0234c5
·
1 Parent(s): 9d3bde8

Fix model links

Browse files
demos/ort-sdturbo/index.html CHANGED
@@ -279,17 +279,17 @@ fn main(@location(0) fragUV : vec2<f32>) -> @location(0) vec4<f32> {
279
 
280
  const models = {
281
  "unet": {
282
- url: "unet.onnx", size: 640,
283
  // should have 'steps: 1' but will fail to create the session
284
  opt: { freeDimensionOverrides: { batch_size: 1, num_channels: 4, height: 64, width: 64, sequence_length: 77, } }
285
  },
286
  "text_encoder": {
287
- url: "text_encoder.onnx", size: 1700,
288
  // should have 'sequence_length: 77' but produces a bad image
289
  opt: { freeDimensionOverrides: { batch_size: 1, } },
290
  },
291
  "vae_decoder": {
292
- url: "vae_decoder.onnx", size: 95,
293
  opt: { freeDimensionOverrides: { batch_size: 1, num_channels_latent: 4, height_latent: 64, width_latent: 64 } }
294
  }
295
  }
 
279
 
280
  const models = {
281
  "unet": {
282
+ url: "https://huggingface.co/webai-community/models/resolve/main/sdturbo-unet.onnx", size: 640,
283
  // should have 'steps: 1' but will fail to create the session
284
  opt: { freeDimensionOverrides: { batch_size: 1, num_channels: 4, height: 64, width: 64, sequence_length: 77, } }
285
  },
286
  "text_encoder": {
287
+ url: "https://huggingface.co/webai-community/models/resolve/main/sdturbo-text-encoder.onnx", size: 1700,
288
  // should have 'sequence_length: 77' but produces a bad image
289
  opt: { freeDimensionOverrides: { batch_size: 1, } },
290
  },
291
  "vae_decoder": {
292
+ url: "https://huggingface.co/webai-community/models/resolve/main/sdturbo-vae-decoder.onnx", size: 95,
293
  opt: { freeDimensionOverrides: { batch_size: 1, num_channels_latent: 4, height_latent: 64, width_latent: 64 } }
294
  }
295
  }
demos/tflite-gemma/index.js CHANGED
@@ -21,7 +21,7 @@ const output = document.getElementById('output');
21
  const submit = document.getElementById('submit');
22
  const status = document.getElementById('status');
23
 
24
- const modelFileName = 'gemma-2b-it-gpu-int4.bin'; /* Update the file name */
25
  let startTime;
26
 
27
  /**
 
21
  const submit = document.getElementById('submit');
22
  const status = document.getElementById('status');
23
 
24
+ const modelFileName = 'https://huggingface.co/webai-community/models/resolve/main/gemma-2b-it-gpu-int4.bin'; /* Update the file name */
25
  let startTime;
26
 
27
  /**