24.06版本网址
仓库 |
Gitee |
GitHub |
DeepSpark |
||
DeepSparkHub |
||
DeepSparkInference |
DeepSparkHub
模型与算法
- 新增了7个PyTorch训练模型和对OpenPCDet工具箱的支持
- 新增了8个大语言模型相关训练示例。
- Aquila2-34B (Megatron-DeepSpeed)
- Baichuan2-7B SFT (DeepSpeed)
- Bloom-7B1 SFT LoRA (Firefly)
- ChatGLM2-6B SFT (DeepSpeed)
- Llama2-7B RLHF (Megatron-DeepSpeed)
- Llama2-13B (Megatron-DeepSpeed)
- Llama2-34B (Megatron-DeepSpeed)
- Qwen-7B SFT LoRA Ptuning-V2 (Firefly)
问题修复
- 修复了CenterNet PyTorch模型的DCNv2 版本不配套的问题。
- 修复了BERT Pytorch模型的h5py依赖库版本不对的问题。
- 修复了YOLOv7模型在PyTorch 2.x版本训练启动报错的问题。
- 同步了YOLOv5,FaceNet,AlexNet,InceptionV3,VGG16模型的代码。
- 完善了BiSeNetV2,DeepLabV3,SSD,YOLOv5模型的说明文档。
感谢以下社区贡献者
majorli,Jino Yang,李一力,shengquan.nian,wenfeng.zhang,吴永乐,zhaojun0044,xiaomei.wang,gaiqin.bai,sanghui-ilu。
DeepSparkInference
模型与算法
- 新增了31个推理小模型示例,其中支持IGIE推理引擎的16个,支持IxRT推理引擎的15个。
- 新增了4个大语言模型的推理示例。
- Baichuan2-7B (vLLM)
- ChatGLM-3-6B (vLLM)
- Llama2-7B (TensorRT-LLM)
- Qwen-7B (Text Generation Inference)
问题修复
- 修复了YOLOX IxRT 插件编译会报错的问题。
- 完善了libGL依赖在Ubuntu上安装的帮助说明。
感谢以下社区贡献者
YoungPeng,majorli,xinchi.tian,tianxi-yi,may,xiaomei.wang,cheneychen2023,qiang.zhang。