Spaces:
Running
Running
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: "
|
288 |
// should have 'sequence_length: 77' but produces a bad image
|
289 |
opt: { freeDimensionOverrides: { batch_size: 1, } },
|
290 |
},
|
291 |
"vae_decoder": {
|
292 |
-
url: "
|
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 |
/**
|