Update README.md

增加背景及相关模型描述
This commit is contained in:
Zhimin Sun 2024-01-10 10:04:29 +08:00 committed by GitHub
parent ca4a9ad632
commit 4a3765c325
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,2 +1,13 @@
# AIPentest
本项目旨在通过使用LLM辅助渗透测试。
初步的思路是基于优选的技术资料做RAG同时逐步外接kali的功能做成渗透测试的agents,表现形式为chatbot.
开发上考虑到便利性先期使用本地模型以fastchat openai兼容接品使用langchain作为开发框架。后续可以方便的切换到openai或其它任意模型。
开发过程中使用的模型为chinese-alpaca2-7b来自 https://github.com/ymcui/Chinese-LLaMA-Alpaca-2
embedding模型为BAAI/bge-large-zh, 来自 https://huggingface.co/BAAI/bge-large-zh/tree/main
fastchat 相关启动脚本见 start
对使用到的相关开源致以感谢。
LLM can support pentest, step by step