Got an error when I used lora to generate images.

[wen020]

Hi, I use code(https://github.com/Stability-AI/StableCascade/blob/master/inference/lora.ipynb) to generate images by lora, but the following error will be reported.

I understand that this is caused by the model not being downloaded correctly. The code I used to download the model is as follows. Is there any problem with this operation?

download models :bash download_models.sh essential small-small bfloat16

[Deeps358]

Hey!

Try to replace model version and generator_checkpoint_path in config for small models.

For small stage B model (stage_b_3b.yaml): model_version: 700M generator_checkpoint_path: models/stage_b_lite_bf16.safetensors

Note: lora parameters may be different, try to look at main readme file, this example for common stage_c For small stage C model (stage_c_3b.yaml): model_version: 1B generator_checkpoint_path: models/stage_c_lite_bf16.safetensors

Also look at #4