diffusion_rs_common-0.1.0
- diffusion_rs_common 0.1.0
- Docs.rs crate page
- MIT
- Links
- Homepage
- Repository
- crates.io
- Source
- Owners
- EricLBuehler
- Dependencies
- accelerate-src ^0.3.2 normal optional
  
  anyhow ^1.0.94 normal
  
  byteorder ^1.5.0 normal
  
  cudarc ^0.12.1 normal optional
  
  dirs ^5.0.1 normal
  
  float8 ^0.1.0 normal
  
  gemm ^0.17.0 normal
  
  half ^2.4.1 normal
  
  hf-hub ^0.3.2 normal
  
  indicatif ^0.17.9 normal
  
  intel-mkl-src ^0.8.1 normal optional
  
  libc ^0.2.147 normal optional
  
  memmap2 ^0.9.5 normal
  
  metal ^0.27.0 normal optional
  
  num-traits ^0.2.15 normal
  
  num_cpus ^1.15.0 normal
  
  rand ^0.8.5 normal
  
  rand_distr ^0.4.3 normal
  
  rayon ^1.1.0 normal
  
  safetensors ^0.4.1 normal
  
  serde ^1.0.215 normal
  
  serde_json ^1.0.133 normal
  
  thiserror ^2.0.4 normal
  
  tokenizers ^0.21.0 normal
  
  tqdm ^0.7.0 normal
  
  ug ^0.0.2 normal
  
  ug-cuda ^0.0.2 normal optional
  
  ug-metal ^0.0.2 normal optional
  
  yoke ^0.7.2 normal
  
  zip ^2.2.1 normal
  
  bindgen_cuda ^0.1.5 build optional
- Versions
- 17.76% of the crate is documented
Platform
- x86_64-unknown-linux-gnu
Feature flags

docs.rs

Rust

rope

diffusion_rs_common0.1.0

In diffusion_rs_common::nn::rotary_emb

diffusion_rs_common::nn::rotary_emb

Function rope

pub fn rope(xs: &Tensor, cos: &Tensor, sin: &Tensor) -> Result<Tensor>