feat: update workspace paths and enhance gitignore
- 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
This commit is contained in:
@@ -0,0 +1,12 @@
|
||||
extern crate alloc;
|
||||
|
||||
#[test]
|
||||
fn test_burnpack_no_std() {
|
||||
use burn_ndarray::NdArray;
|
||||
use burn_no_std_tests::burnpack;
|
||||
type Backend = NdArray<f32>;
|
||||
let device = Default::default();
|
||||
|
||||
// Run all Burnpack tests
|
||||
burnpack::run_all_tests::<Backend>(&device);
|
||||
}
|
||||
@@ -0,0 +1,12 @@
|
||||
extern crate alloc;
|
||||
|
||||
#[test]
|
||||
fn test_safetensors_no_std() {
|
||||
use burn_ndarray::NdArray;
|
||||
use burn_no_std_tests::safetensors;
|
||||
type Backend = NdArray<f32>;
|
||||
let device = Default::default();
|
||||
|
||||
// Run all SafeTensors tests
|
||||
safetensors::run_all_tests::<Backend>(&device);
|
||||
}
|
||||
@@ -0,0 +1,31 @@
|
||||
#![no_std] // Must keep it for testing
|
||||
|
||||
use burn_no_std_tests::mlp::*;
|
||||
use burn_no_std_tests::model::*;
|
||||
|
||||
use burn::tensor::{Distribution, Tensor, backend::Backend};
|
||||
use burn_ndarray::NdArray;
|
||||
|
||||
#[test]
|
||||
fn test_mnist_model_with_random_input() {
|
||||
type Backend = NdArray<f32>;
|
||||
|
||||
// Model configurations
|
||||
let device = Default::default();
|
||||
let mlp_config = MlpConfig::new();
|
||||
let mnist_config = MnistConfig::new(mlp_config);
|
||||
let mnist_model: Model<Backend> = Model::new(&mnist_config, &device);
|
||||
|
||||
// Pass a fixed seed for random, otherwise a build generated random seed is used
|
||||
Backend::seed(&device, mnist_config.seed);
|
||||
|
||||
// Some random input
|
||||
let input_shape = [1, 28, 28];
|
||||
let input = Tensor::<Backend, 3>::random(input_shape, Distribution::Default, &device);
|
||||
|
||||
// Run through the model
|
||||
let output = mnist_model.forward(input);
|
||||
|
||||
assert_eq!(&*output.shape(), [1, 10]);
|
||||
assert!(output.to_data().iter::<f32>().all(|x| x <= 1.0));
|
||||
}
|
||||
Reference in New Issue
Block a user