mirror of
https://github.com/bentoml/OpenLLM.git
synced 2026-05-04 05:42:44 -04:00
820b4991fa
This is to help with working on CPU machine Signed-off-by: aarnphm-ec2-dev <29749331+aarnphm@users.noreply.github.com>
4 lines
144 B
Python
Generated
4 lines
144 B
Python
Generated
import torch
|
|
|
|
def rotary_embedding_neox(arg0: torch.Tensor, arg1: torch.Tensor, arg2: torch.Tensor, arg3: int, arg4: torch.Tensor) -> None: ...
|