lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-02-06 16:09:58 +00:00
Code Issues Packages Projects Releases Wiki Activity

UNet from Scratch

Browse Source 
Now backend rewrite is about 50% finished.
Estimated finish is in 72 hours.
After that, many newer features will land.
This commit is contained in:
layerdiffusion
2024-08-01 21:19:41 -07:00
parent e3522c8919
commit bc9977a305
20 changed files with 1393 additions and 56 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
extensions-builtin/forge_preprocessor_reference/scripts/forge_reference.py
View File

@@ -61,14 +61,14 @@ class PreprocessorReference(Preprocessor):
# This is a powerful VAE with integrated memory management, bf16, and tiled fallback.
latent_image = vae.encode(cond.movedim(1, -1))
latent_image = process.sd_model.forge_objects.unet.model.latent_format.process_in(latent_image)
latent_image = process.sd_model.forge_objects.vae.first_stage_model.process_in(latent_image)
gen_seed = process.seeds[0] + 1
gen_cpu = torch.Generator().manual_seed(gen_seed)
unet = process.sd_model.forge_objects.unet.clone()
sigma_max = unet.model.model_sampling.percent_to_sigma(start_percent)
sigma_min = unet.model.model_sampling.percent_to_sigma(end_percent)
sigma_max = unet.model.prediction.percent_to_sigma(start_percent)
sigma_min = unet.model.prediction.percent_to_sigma(end_percent)
self.recorded_attn1 = {}
self.recorded_h = {}