RunPod•4mo ago

SDXL Dreambooth - can’t load model

Hi there, Does anyone know why my script is not loading this SDXL model?

Solution:

Ah apparently, the script that I am running, is only for SD1.5

Jump to solution

28 Replies

nerdylive•4mo ago

Check the model format Is it .sa It should be .safetensors

smoke•4mo ago

Noo it’s not, I’m not sure why it is split in the logs I checked through the web terminal and the file is there

nerdylive•4mo ago

smoke•4mo ago

~6 GB

nerdylive•4mo ago

Can you take a ss Just to make sure

smoke•4mo ago

It’s the second one It seems like it can find the checkpoint but not loading it for some reason

nerdylive•4mo ago

Ooh it might be corrupted.. maybe How did you download it

smoke•4mo ago

It gets downloaded inside my Dockerfile 1 sec I’ll share the lines

nerdylive•4mo ago

Ooh It says there's an error in the model loading State dict, I'm not sure what does it means either

Madiator2011 (Work)•4mo ago

I'm not sure if DB endpoint supports SDXL as it's diffrent arch than SD 1.5 and 2.1

nerdylive•4mo ago

oh dreambooth i thought db=database mb lol

Madiator2011 (Work)•4mo ago

It's serverless channel so I though question is about serverless

nerdylive•4mo ago

yeahh

Madiator2011 (Work)•4mo ago

though it looks like our friend runs on pod

nerdylive•4mo ago

yea maybe try kohyaa trainer smoke

smoke•4mo ago

It is definitely serverless yeah Dreambooth should work with SDXL though And this is using sd-scripts which Kohya uses under the hood

nerdylive•4mo ago

Ohh Soo.. How did you download it Try searching that problem your experiencing on Google too Maybe some part of the logs match with someone's problem

smoke•4mo ago

I have this line in my Dockerfile so two models get baked into the image

RUN mkdir -p /model_cache && \
    wget "https://civitai.com/api/download/models/292213?type=Model&format=SafeTensor&size=full&fp=fp16&token=${CIVITAI_TOKEN}" -O /model_cache/hyperRealism_30.safetensors && \
    wget "https://civitai.com/api/download/models/344487?type=Model&format=SafeTensor&size=pruned&fp=fp16&token=${CIVITAI_TOKEN}" -O /model_cache/realvisxlV40_v40.safetensors

RUN mkdir -p /model_cache && \
    wget "https://civitai.com/api/download/models/292213?type=Model&format=SafeTensor&size=full&fp=fp16&token=${CIVITAI_TOKEN}" -O /model_cache/hyperRealism_30.safetensors && \
    wget "https://civitai.com/api/download/models/344487?type=Model&format=SafeTensor&size=pruned&fp=fp16&token=${CIVITAI_TOKEN}" -O /model_cache/realvisxlV40_v40.safetensors

nerdylive•4mo ago

Oh should be good

smoke•4mo ago

For some reason the shape is different or something

size mismatch for down_blocks.1.attentions.0.proj_in.weight: copying a param with shape torch.Size([640, 640]) from checkpoint, the shape in current model is torch.Size([640, 640, 1, 1]).

size mismatch for down_blocks.1.attentions.0.proj_in.weight: copying a param with shape torch.Size([640, 640]) from checkpoint, the shape in current model is torch.Size([640, 640, 1, 1]).

nerdylive•4mo ago

Hmm I have no clue for that

Madiator2011 (Work)•4mo ago

mayby it's fp16

smoke•4mo ago

I think it is yes

nerdylive•4mo ago

So what's the normal one for dream booth? It's not supported?

Solution

smoke•4mo ago

Ah apparently, the script that I am running, is only for SD1.5

smoke•4mo ago

This is the source code of kohya I have to run a different script for SDXL apparently. Sorry guys, missed that one

nerdylive•4mo ago

Haha alright, nice

Gaming

Programming

SDXL Dreambooth - can’t load model