HubAi dont return image in production with @cf/bytedance/stable-diffusion-xl-lightning (in dev it works)

[samulefevre]

const res = await hubAI().run('@cf/bytedance/stable-diffusion-xl-lightning', {
   prompt,
}) // return Uint8Array
// seems like in production it don't return the image (in dev i get image)

const blob = new Blob([res])

// then save image with hubBlob() with blob
// in dev i see image in blobs, in prod i got empty image(15B)

But why in cloudflare account, it say i use @cf/baai/bge-small-en-v1.5

(it works in dev, i got image but in prod no image)

i tried with other model : @cf/stabilityai/stable-diffusion-xl-base-1.0. same result works in dev but not working in production, what the hell is happening ?

[atinux]

Do you have a small reproduction I can investigate @samulefevre ?

On solution it to make sure to transform as blob only when it's not one:

let blob = await hubAI().run('@cf/bytedance/stable-diffusion-xl-lightning', {
   prompt,
})
if (blob instanceof Uint8Array) {
  blob = new Blob([blob])
}

I also have some @cf/baai/bge-small in my account on Cloudflare but I don't think it's related as beta models are not displayed in their dashboard.

[samulefevre]

It works with your solution, thank you !

[atinux]

Pushed a fix in v0.7.7 so you don't have to do this anymore normally 😊

[samulefevre]

Nice ! Thank you. Tested in preview, it works !