RunPod•5mo 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•5mo ago

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

smokeOP•5mo 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•5mo ago

smokeOP•5mo ago

~6 GB

nerdylive•5mo ago

Can you take a ss Just to make sure

smokeOP•5mo ago

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

nerdylive•5mo ago

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

smokeOP•5mo ago

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

nerdylive•5mo 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)•5mo ago

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

nerdylive•5mo ago

oh dreambooth i thought db=database mb lol

Madiator2011 (Work)•5mo ago

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

nerdylive•5mo ago

yeahh

Madiator2011 (Work)•5mo ago

though it looks like our friend runs on pod

nerdylive•5mo ago

yea maybe try kohyaa trainer smoke

smokeOP•5mo 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•5mo 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

smokeOP•5mo 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•5mo ago

Oh should be good

smokeOP•5mo 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•5mo ago

Hmm I have no clue for that

Madiator2011 (Work)•5mo ago

mayby it's fp16

smokeOP•5mo ago

I think it is yes

nerdylive•5mo ago

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

Solution

smoke•5mo ago

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

smokeOP•5mo ago

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

nerdylive•5mo ago

Haha alright, nice

Gaming

Programming

SDXL Dreambooth - can’t load model