Git Product home page Git Product logo

Comments (7)

jang11441211 avatar jang11441211 commented on May 31, 2024
head.weight                       f16   cuda:0   2048 65536
D:\RWKV\backend-python\rwkv_pip\model.py:1824: UserWarning: operator () profile_node %318 : int = prim::profile_ivalue(%316)
 does not have profile information (Triggered internally at ..\torch\csrc\jit\codegen\cuda\graph_fuser.cpp:109.)
  r, k, v, g, xxx, ss = self.v5_2_before(
INFO:     127.0.0.1:49843 - "POST /switch-model HTTP/1.1" 200 OK
INFO:     127.0.0.1:50019 - "OPTIONS /v1/chat/completions HTTP/1.1" 200 OK
INFO:     127.0.0.1:50019 - "POST /v1/chat/completions HTTP/1.1" 200 OK
ERROR:    Exception in ASGI application
Traceback (most recent call last):
  File "D:\RWKV\py310\Lib\site-packages\uvicorn\protocols\http\h11_impl.py", line 408, in run_asgi
    result = await app(  # type: ignore[func-returns-value]
  File "D:\RWKV\py310\Lib\site-packages\uvicorn\middleware\proxy_headers.py", line 84, in __call__
    return await self.app(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\fastapi\applications.py", line 1115, in __call__
    await super().__call__(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\applications.py", line 122, in __call__
    await self.middleware_stack(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\middleware\errors.py", line 184, in __call__
    raise exc
  File "D:\RWKV\py310\Lib\site-packages\starlette\middleware\errors.py", line 162, in __call__
    await self.app(scope, receive, _send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\middleware\cors.py", line 91, in __call__
    await self.simple_response(scope, receive, send, request_headers=headers)
  File "D:\RWKV\py310\Lib\site-packages\starlette\middleware\cors.py", line 146, in simple_response
    await self.app(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\middleware\exceptions.py", line 79, in __call__
    raise exc
  File "D:\RWKV\py310\Lib\site-packages\starlette\middleware\exceptions.py", line 68, in __call__
    await self.app(scope, receive, sender)
  File "D:\RWKV\py310\Lib\site-packages\fastapi\middleware\asyncexitstack.py", line 20, in __call__
    raise e
  File "D:\RWKV\py310\Lib\site-packages\fastapi\middleware\asyncexitstack.py", line 17, in __call__
    await self.app(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\routing.py", line 718, in __call__
    await route.handle(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\routing.py", line 276, in handle
    await self.app(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\starlette\routing.py", line 69, in app
    await response(scope, receive, send)
  File "D:\RWKV\py310\Lib\site-packages\sse_starlette\sse.py", line 233, in __call__
    async with anyio.create_task_group() as task_group:
  File "D:\RWKV\py310\Lib\site-packages\anyio\_backends\_asyncio.py", line 597, in __aexit__
    raise exceptions[0]
  File "D:\RWKV\py310\Lib\site-packages\sse_starlette\sse.py", line 236, in wrap
    await func()
  File "D:\RWKV\py310\Lib\site-packages\sse_starlette\sse.py", line 221, in stream_response
    async for data in self.body_iterator:
  File "D:\RWKV\backend-python\routes\completion.py", line 149, in eval_rwkv
    for response, delta, prompt_tokens, completion_tokens in model.generate(
  File "D:\RWKV\backend-python\utils\rwkv.py", line 270, in generate
    token = self.pipeline.sample_logits(
  File "D:\RWKV\backend-python\rwkv_pip\utils.py", line 143, in sample_logits
    out = torch.multinomial(probs, num_samples=1)[0]
RuntimeError: probability tensor contains either `inf`, `nan` or element < 0

from rwkv-runner.

josStorer avatar josStorer commented on May 31, 2024

配置页面截图看一下

from rwkv-runner.

jang11441211 avatar jang11441211 commented on May 31, 2024

image

from rwkv-runner.

josStorer avatar josStorer commented on May 31, 2024

什么显卡, 试试关掉自定义CUDA算子

from rwkv-runner.

jang11441211 avatar jang11441211 commented on May 31, 2024

显卡是1650,我关闭cuda后。可以正常输出了。谢谢您的帮助

from rwkv-runner.

josStorer avatar josStorer commented on May 31, 2024

可以nvidia-smi看看驱动版本, 尝试更新驱动, 自定义算子能开最好开, 虽然1.5B不开速度也可以

from rwkv-runner.

josStorer avatar josStorer commented on May 31, 2024

此外你可以试试WebGPU(Python)模式, 开nf4跑3B, 或许会比1.5B的效果更好

from rwkv-runner.

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.