search

rmusab / vul-llm-finetune

Finetuning large language models (LLMs) for vulnerability detection
34 stars 5 forks source link

The fine-tuned WizardCoder model #2

Open Yuuoniy opened 6 months ago

Yuuoniy commented 6 months ago

Dear authors, Thanks for your work! I am interested in applying it in my study. I wonder could you provide the fine-tuned WizardCoder model file, which could be ready-for-use. Or could you please provide a detailed instruction for fine-tuning the WizardCoder model. I couldn't see any scripts related to the WizardCoder model in current repo. Thank you in advance!

Adefful commented 4 months ago

Thank you for your interest in our work. You can fine-tune WizardCoder using the scripts for StarCoder, as WizardCoder is based on StarCoder. You can find the training code in our repository at:

https://github.com/rmusab/vul-llm-finetune/blob/main/LLM/starcoder/finetune/run.py

If you need further assistance or have any questions about the process, please don't hesitate to ask.