- Updated stablediffusion crate path from "../stable-diffusion-burn" to "./crates/stable-diffusion-burn" for proper workspace resolution - Enhanced .gitignore to include generated model files (.mpk, .pt, .bin, .safetensors, .ckpt) and user_data directory - Added Cargo.lock to gitignore with appropriate comment - Reorganized IDE files section in gitignore for better clarity - Added newline at end of file for proper formatting
75 lines
1.0 KiB
Rust
75 lines
1.0 KiB
Rust
#[allow(unused_imports)] // required for re-included modules
|
|
pub use super::*;
|
|
|
|
mod abs;
|
|
mod adaptive_avgpool1d;
|
|
mod adaptive_avgpool2d;
|
|
mod add;
|
|
mod aggregation;
|
|
mod avgpool1d;
|
|
mod avgpool2d;
|
|
mod backward;
|
|
mod bridge;
|
|
mod broadcast;
|
|
mod cast;
|
|
mod cat;
|
|
mod ceil;
|
|
mod checkpoint;
|
|
mod complex;
|
|
mod conv1d;
|
|
mod conv2d;
|
|
mod conv3d;
|
|
mod conv_transpose1d;
|
|
mod conv_transpose2d;
|
|
mod conv_transpose3d;
|
|
mod cross;
|
|
mod cross_entropy;
|
|
mod cummax;
|
|
mod cummin;
|
|
mod cumprod;
|
|
mod cumsum;
|
|
mod deform_conv2d;
|
|
mod div;
|
|
mod erf;
|
|
mod exp;
|
|
mod expand;
|
|
mod flip;
|
|
mod floor;
|
|
mod gather_scatter;
|
|
mod gelu;
|
|
mod gradients;
|
|
mod log;
|
|
mod log1p;
|
|
mod log_sigmoid;
|
|
mod mask;
|
|
mod matmul;
|
|
mod maxmin;
|
|
mod maxpool1d;
|
|
mod maxpool2d;
|
|
mod memory_management;
|
|
mod mul;
|
|
mod multithread;
|
|
mod nearest_interpolate;
|
|
mod neg;
|
|
mod nonzero;
|
|
mod permute;
|
|
mod pow;
|
|
mod recip;
|
|
mod relu;
|
|
mod remainder;
|
|
mod repeat_dim;
|
|
mod reshape;
|
|
mod round;
|
|
mod select;
|
|
mod sigmoid;
|
|
mod sign;
|
|
mod slice;
|
|
mod slice_assign;
|
|
mod softmax;
|
|
mod sort;
|
|
mod sqrt;
|
|
mod sub;
|
|
mod transpose;
|
|
mod trig;
|
|
mod unfold;
|