Skip to content

Latest commit

 

History

History
72 lines (48 loc) · 3.76 KB

File metadata and controls

72 lines (48 loc) · 3.76 KB

认领者 GitHub ID

megemini

赛题信息

  • 进阶任务序号:#15
  • 赛题名称:基于天数智芯硬件与文心多模态模型的创新应用
  • 关联厂商:天数

本周工作

  1. RFC 文档

  2. 代码实现

    • 已经完成 AI Studio 项目的 notebook
    • 已经创建了双卡的天数环境
    • 已完成 cli 的脚本,drug_ocr_cli.py 针对最新实现的脚本,有两处大的改进:
      1. 增加了 patch aistudio sdk 的函数,原因是,paddlespeech 与 fastdeploy 对于 aistudio sdk 使用的版本不一样,因此,需要先利用脚本修改源码,将其统一
      2. 增加了针对 tts 合成的音频文件进行音量修改的函数,原因是,天数的框架开发环境的音频编码库好像与 paddlespeech 有点兼容问题,导致合成的音频会出现截止的情况,因此,与 paddlespeech 的研发讨论后,决定增加这个后处理的函数。
    • 已发布 AI Studio notebook 项目:https://aistudio.baidu.com/projectdetail/10413884

    注意:因为后面提到的 AI Studio 环境问题,此 notebook 的 ERNIE-4.5-0.3B-Paddle 输出混乱,因此,此 notebook 仅作为参考,可在本地最新的天数环境运行调试。

    • 已经在天数双卡环境中验证了 tensor_parallel_size2 时,可以加载与使用 ERNIE-4.5-VL-28B-A3B-Thinking 模型,单卡占用显存约 20G ,共计 40G 显存的占用。
  3. README

    • 可以参考 AI Studio 项目的 notebook
  4. 演示视频/截图

    ernie28b

  5. 问题与解决

    • 问题:天数的框架开发环境极度不稳定,导致没有办法持续的验证优化后的脚本。

    天数的框架开发环境好像是共享的模式,所谓的启动、关闭,只是用于控制用户是否可以 ssh 远程连接到服务器。这就导致了,经常出现:

    • 突然被断开连接,踢出了环境
    • 新连接的环境,显存已经被占用满了
    • 运行过程中提示硬盘没有空间了,实际上 aistudio 的工作目录只有 76G 的文件(包括模型文件等)
    • 运行过程中提示识别不到模型,实际上模型没有问题,可能再运行一次就好了
    • 运行过程中加载模型很慢,有的时候要将近10分钟才能加载完 ERNIE-4.5-VL-28B-A3B-Thinking
    • 加载完模型后,输出 token 到一半就卡住了,再运行一次可能又会在其他地方卡住
    • ixsmi 命令有时候不能反应当前环境的显存使用情况,比如,模型都加载完了,还显示只有 64MB 的显存占用

    以上只是这两周遇到的部分环境问题,导致,从上周开始调试到现在,只有 2 次能够完整的运行完脚本,其他时间都是不断的被各种情况打断。

    目前的状态是:脚本应该没有什么问题,但是,还需要再至少完整的运行完一次,从而抓取完整的日志。

    周报目录中的 drug_ernie03.log 是使用 ERNIE-4.5-0.3B-Paddle 完整运行后的日志,ERNIE-4.5-VL-28B-A3B-Thinking 也完整运行过一次,不过,当时并没有注意到这个模型会先输出 thinking 部分,导致最终输出的 token 不够,因此,最新的 drug_ocr_cli.py 脚本已经进行了修改,但是,到目前为止,整整一周多的时间都没有再完整的运行过一次。

下周计划

  1. 调试双卡环境

当前阻塞(无则填"无")

  • AI Studio 环境极度不稳定

交付物进展

交付物 状态 备注
RFC 文档 ✅ 已完成 -
代码实现 ✅ 已完成
README ✅ 已完成 -
演示视频/截图 ✅ 已完成 -