search

GreyDGL / PentestGPT

A GPT-empowered penetration testing tool
MIT License
6.98k stars 835 forks source link

Question - Any plans to support local LLM models? #124

Closed 0xm4v3rick closed 6 months ago

0xm4v3rick commented 1 year ago

Hi Team,

Thanks for making this tool. Are there any plans to support local LLM model for this as an alternative to GPT? Pentest contains sensitive info and it would be good to not share this with a SaaS solution and keep it local. An configuration option to have connect to local model would be great to have.

Thanks!

busraoguzoglu commented 1 year ago

Hello, is there any progress on this? Is it possible to use the tool with other models like LLAMA? I saw that the latest update made this possible but will you add any examples for that?

GreyDGL commented 1 year ago

@busraoguzoglu Yes. Please check the latest commit. We have supported GPT4All models. Note that during the testing, we notice that all the open-source models are not as good as GPT4.

SATUNIX commented 12 months ago

@busraoguzoglu Yes. Please check the latest commit. We have supported GPT4All models. Note that during the testing, we notice that all the open-source models are not as good as GPT4.

Good point, its best to stick with GPT4API For now until specific models can be trained I've ran some testing using Vicuna and LLaMa2 and the results were negligible..

GreyDGL commented 12 months ago

@SATUNIX fine-tuned Vicuna-13b looks fine, but other than that, all the open-source models below 13B are kinda not working at all.

GreyDGL commented 6 months ago

I'll merge this issue to #69 and implement further features.