fixed the missing changes

JulesBelveze / bert-squeeze

🛠️ Tools for Transformers compression using PyTorch Lightning ⚡

https://julesbelveze.github.io/bert-squeeze/

78 stars 10 forks source link

Closed tarekziade closed 5 months ago

tarekziade commented 5 months ago

Few fixes:

The examples were breaking because pretrained_model_name was not renamed to pretrained_model_name_or_path everyhere
the target lookup at instanciation can break if the key is missing (using the example config)
fixed the pads in the doc