Foreign-Mirrors/Fooocus
1
0
Fork 0
mirror of https://github.com/lllyasviel/Fooocus.git synced 2026-08-16 13:13:16 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix vae

Browse Source
This commit is contained in:
lvmin
2023-10-14 04:39:56 -07:00
parent 42c7365103
commit 79b1a605a7
1 changed files with 10 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/default_pipeline.py
+2 -6
View File
@@ -452,16 +452,12 @@ def process_diffusion(positive_cond, negative_cond, steps, switch, width, height
target_model = final_unet target_model = final_unet
print('Use base model to refine itself - this may because of developer mode.') print('Use base model to refine itself - this may because of developer mode.')
sigmas = None
len_sigmas = steps - switch
if final_refiner_unet is not None:
sigmas = calculate_sigmas(sampler=sampler_name, sigmas = calculate_sigmas(sampler=sampler_name,
scheduler=scheduler_name, scheduler=scheduler_name,
model=final_refiner_unet.model, model=target_model.model,
steps=steps, steps=steps,
denoise=denoise)[switch:] denoise=denoise)[switch:]
k1 = final_refiner_unet.model.latent_format.scale_factor k1 = target_model.model.latent_format.scale_factor
k2 = final_unet.model.latent_format.scale_factor k2 = final_unet.model.latent_format.scale_factor
k = float(k1) / float(k2) k = float(k1) / float(k2)
sigmas = sigmas * k sigmas = sigmas * k