From 2b7c00d3d2d9f751258d552d09f883a92f505eea Mon Sep 17 00:00:00 2001 From: Jakob Sachs Date: Wed, 28 Jan 2026 15:07:19 +0100 Subject: [PATCH] fix sd-example dtype for CLIP embeddings (#14397) --- examples/webgpu/stable_diffusion/compile.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/webgpu/stable_diffusion/compile.py b/examples/webgpu/stable_diffusion/compile.py index 6f47a5b3c6..cfcbe837f2 100644 --- a/examples/webgpu/stable_diffusion/compile.py +++ b/examples/webgpu/stable_diffusion/compile.py @@ -93,7 +93,7 @@ if __name__ == "__main__": forward: Any = None sub_steps = [ - Step(name = "textModel", input = [Tensor.randn(1, 77)], forward = model.cond_stage_model.transformer.text_model), + Step(name = "textModel", input = [Tensor.randint(1, 77, low=0, high=49408, dtype=dtypes.int32)], forward = model.cond_stage_model.transformer.text_model), Step(name = "diffusor", input = [Tensor.randn(1, 77, 768), Tensor.randn(1, 77, 768), Tensor.randn(1,4,64,64), Tensor.rand(1), Tensor.randn(1), Tensor.randn(1), Tensor.randn(1)], forward = model), Step(name = "decoder", input = [Tensor.randn(1,4,64,64)], forward = model.decode), Step(name = "f16tof32", input = [Tensor.randn(2097120, dtype=dtypes.uint32)], forward = u32_to_f16)