load transformer model error

TheDenk / cogvideox-controlnet

Simple Controlnet module for CogvideoX model.

Apache License 2.0

64 stars 2 forks source link

load transformer model error #8

Open linwenzhao1 opened 1 week ago

linwenzhao1 commented 1 week ago

The format for loading a model into the code is Pt. I got the following error while executing the inference code. How to solve it?

TheDenk commented 1 week ago

I used Pt format for simple tuning mode (for me .pt is simpler for model training) you can change code in condition after if args.pretrained_controlnet_path: to load .safetensors model. For example as I did in jupyter notebook example.

Also you provide too small log info, I only can say that model path is incorrect)

linwenzhao1 commented 1 week ago

I use my own data to fine-tune. but the reasoning result is incorrect. Do you know the reason?

TheDenk commented 4 days ago

Sorry, I don't know why this is happening.