burn_tripo 0.1.0

TripoSG(-scribble) implemented in burn
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26

use burn::prelude::*;

#[cfg(target_arch = "wasm32")]
fn readback_not_supported() -> String {
    "tensor readback requires async handling on wasm32; \
     use an async readback path or disable GPU readback for this build"
        .to_string()
}

#[cfg(not(target_arch = "wasm32"))]
pub(crate) fn tensor_to_vec_f32<B: Backend, const D: usize>(
    tensor: Tensor<B, D>,
) -> Result<Vec<f32>, String> {
    tensor
        .into_data()
        .convert::<f32>()
        .to_vec::<f32>()
        .map_err(|_| "failed to read tensor data".to_string())
}

#[cfg(target_arch = "wasm32")]
pub(crate) fn tensor_to_vec_f32<B: Backend, const D: usize>(
    _tensor: Tensor<B, D>,
) -> Result<Vec<f32>, String> {
    Err(readback_not_supported())
}