search

ostris / ai-toolkit

Various AI scripts. Mostly Stable Diffusion stuff.
MIT License
220 stars 25 forks source link

Can't train slider on NAI #17

Closed Poiuytrezay1 closed 9 months ago

Poiuytrezay1 commented 9 months ago

I have tried running the slider script on NAI models, but I get an error every time. The example worked just fine.

On the fp16 safetensors version, I get the error: AttributeError: 'CLIPTextModel' object has no attribute 'embeddings' On the original ckpt version, I get the error: _KeyError: 'timeembed.0.weight'

I've tried changing the dtype, but to no avail. Any chance to fix this?

jaretburkett commented 9 months ago

It is an issue with those models. They use a non standard format or something is corrupted. I am not 100% sure. I load models through diffusers, so their library doesn't support it. I think there are some fixed versions of the models out there that are compatible with diffuser, but not sure where to get them.