Error: failed assertion `length(16640) must be <= 16384.'

[hdsenevi]

I'm getting the following error: failed assertion 'length(16640) must be <= 16384.'

Steps to reproduce :

• Create a new projet
• Add native-diffusion as a dependency (branch: master)
• Download pre-converted bins from here
• Run python web server that serves bin.zip from localhost:8080
• Use SimpleDiffusion example here

I launch the app, I type "astronaut cat" in the TextField. And after a few seconds I got this error :

-[MTLDebugComputeCommandEncoder setThreadgroupMemoryLength:atIndex:]:543: failed assertion `length(16640) must be <= 16384.'

Specs

Tested on
Mac details here

Exported to

• iPhone 14 Pro Max
• iPad Pro 6th gen

Screenshot

Logs

ipad-log.log
iphone-log.log