HackerTrans
TopNewTrendsCommentsPastAskShowJobs

aislamov

no profile record

Submissions

Latent consistency model working directly in the browser on WebGPU

islamov.ai
3 points·by aislamov·3 года назад·1 comments

Stable Diffusion WebGPU demo

islamov.ai
473 points·by aislamov·3 года назад·116 comments

[untitled]

1 points·by aislamov·3 года назад·0 comments

comments

aislamov
·3 года назад·discuss
Model allows image generation in 4-8 steps. 8 steps on M1 Max take 8 seconds, on RTX 3090Ti just 3 seconds. Requires latest chrome canary. Model paper here https://latent-consistency-models.github.io
aislamov
·3 года назад·discuss
> Hold on, to run your demo does one have to click the "Load Model" button before doing anything?

Yes. I thought it won't be good if it would download 3.5gb once you open the page.

>Now I'm wondering whether the top message goes away once the flags are enabled?

No, I haven't added any checks for that (and I'm not sure how the first one can be properly checked), so it's just an info bar. Which is, eventually, misleading.
aislamov
·3 года назад·discuss
What error message do you get?
aislamov
·3 года назад·discuss
UNET takes about a 1:10 on WebGPU and around a minute on CPU in one thread. VAE is 2 minutes on CPU and about 10 seconds on GPU. It should be because most GPU ops for VAE are already implemented but for UNET are not. So in the latter case browser is just tossing data from GPU to CPU and back on each step
aislamov
·3 года назад·discuss
That's really impressive and much more performant. I was following a different approach: to run any ONNX model without prior modifications.
aislamov
·3 года назад·discuss
No additional libraries required on Windows (even CUDA). For linux you'll need to install CUDA and onnxruntime gpu lib.