You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
3 lines
169 B
Python
3 lines
169 B
Python
9 months ago
|
from petals.triton.rmsnorm import rmsnorm_triton_wrapper
|
||
|
from petals.triton.attention import attention_triton_wrapper
|
||
|
from petals.triton.rotary import rbe_triton_wrapper
|