Ben_Kosytorz/burn-stablediffusion-vibecode
1
0
Fork 0
mirror of https://gitea.hainer-ernst.de/rasmus/burn-stablediffusion-vibecode.git synced 2026-06-11 02:09:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add files via upload

Browse Source 
Add initial project files
This commit is contained in:
Gadersd
2023-08-04 14:32:47 -04:00
committed by GitHub
parent 1aed8b655a
commit e4145441eb
31 changed files with 266571 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
python/stablediffusion.py Normal file
View File

@@ -0,0 +1,12 @@
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):
save_scalar(stable_diffusion.alphas_cumprod.shape[0], "n_steps", path)
save_tensor(stable_diffusion.alphas_cumprod, 'alphas_cumprod', path)
save_autoencoder(stable_diffusion.autoencoder, 'autoencoder', path)
save_unet_model(stable_diffusion.diffusion, 'unet', path)
save_clip_text_transformer(stable_diffusion.clip, 'clip', path)