Ben_Kosytorz
3a67c0979c
- 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
14 lines
730 B
Python
14 lines
730 B
Python
import pathlib
|
|
from autoencoder import save_autoencoder
|
|
from unet import save_unet_model
|
|
from clip import save_clip_text_transformer
|
|
|
|
from save import save_scalar, save_tensor
|
|
|
|
def save_stable_diffusion(stable_diffusion, path):
|
|
pathlib.Path(path).mkdir(parents=True, exist_ok=True)
|
|
save_scalar(stable_diffusion.alphas_cumprod.shape[0], "n_steps", path)
|
|
save_tensor(stable_diffusion.alphas_cumprod, 'alphas_cumprod', path)
|
|
save_autoencoder(stable_diffusion.first_stage_model, pathlib.Path(path, 'autoencoder'))
|
|
save_unet_model(stable_diffusion.model.diffusion_model, pathlib.Path(path, 'unet'))
|
|
save_clip_text_transformer(stable_diffusion.cond_stage_model.transformer.text_model, pathlib.Path(path, 'clip')) |