kathrinse / be_great

A novel approach for synthesizing tabular data using pretrained large language models
MIT License
276 stars 46 forks source link

PEFT import fixed #52

Closed erland-ramadhan closed 4 months ago

erland-ramadhan commented 4 months ago

When efficient_finetuning argument is set to lora when initializing the model, it generated an error when importing prepare_model_for_int8_training. So, I tried to change it to prepare_model_for_kbit_training.

Screen Shot 2024-05-30 at 22 12 48