Layers errors (general)

a) You didn’t install the newest patch and still use e.g. beta 2 with some newer model

b) You could check Roformer v2 instead of v1 during installing of the custom model.

c) Model trainer didn't clean the weight using this Python script (you can do it by yourself)

Usage of the script (install Python, and on fresh installation execute “pip install torch” then execute):

python C:\path_of_the_script\prepare_weights_for_inference.py --checkpoint "path_to_the_input_model/model_to_clean.ckpt" --output_file "path_to_the_output_model/cleaned_model.ckpt”

d) You can use MSST instead which doesn't need it