TypeError: Linear.forward() got an unexpected keyword argument 'scale'

[AgnieszkaPolowczyk]

Hi,
I encountered the following error when trying to run training

TypeError: Linear.forward() got an unexpected keyword argument 'scale'
Traceback (most recent call last):
File "training.py", line 37, in
main(conf)
File "training.py", line 16, in main
cfr_lora_training(conf.MACE)
File "src/cfr_lora_training.py", line 522, in main
model_pred = unet(noisy_latents, timesteps, encoder_hidden_states).sample
...
File "src/mace_lora_atten_processor.py", line 43, in call
query = attn.to_q(hidden_states, scale=scale)
TypeError: Linear.forward() got an unexpected keyword argument 'scale'

My environment:
python 3.10.18
diffusers 0.22.0
transformers 4.46.2
huggingface-hub 0.25.2

[Shilin-LU]

Hi, kindly refer to this issue

[AgnieszkaPolowczyk]

Thanks for the reply.
I followed all the suggestions, but I’m still getting the same error:
TypeError: Linear.forward() got an unexpected keyword argument 'scale'.
My environment uses diffusers==0.22.0 transformers==4.38.1 huggingface_hub==0.25.2

Could you please advise what else I could check?

[Shilin-LU]

This issue is likely to be related to version incompatibilities between the Diffusers and Transformers libraries. You might consider reinstalling the environment from scratch and carefully verifying each step during the setup process.