Git Product home page Git Product logo

Comments (1)

lcl6679292 avatar lcl6679292 commented on May 20, 2024

在3090上用cuda跑,在agent的run函数中添加了打印结果和显示时间的代码

generate prompt and call llm

prompt = self.prompt_generator.generate(llm_result, exec_result) start = time.time() llm_result = self.llm.generate(prompt) end = time.time() print(f"===llm_result: {llm_result}, time: {end-start}")

结果发现生成下面几个字符都要20s时间。 ===llm_result: 明天杭州的天气预计是晴天,需要注意防晒哦。, time: 20.340392589569092

性能这么慢不正常吧,完全没法使用,请问可能是什么原因导致。 另外flash_attn安装成功,但是rms_norm 和rotary两个库没有编成功,编译的时候一直卡住。

这个推理时间确实有点异常,可能是环境问题,你这边直接使用modelscope对应模型的pipeline推理需要多久呢,比如这个:

https://modelscope.cn/models/damo/ModelScope-Agent-7B/summary

image

from modelscope-agent.

Related Issues (20)

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.