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: turbo scheduler loading issue (#3065)

Browse Source 
* fix: correctly load ModelPatcher

* feat: do not load model at all, not needed
This commit is contained in:
Manuel Schmid
2024-05-31 22:24:19 +02:00
committed by GitHub
parent 3ef663c5b7
commit 7899261755
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ldm_patched/contrib/external_custom_sampler.py
+1 -2
View File
@@ -107,8 +107,7 @@ class SDTurboScheduler:
def get_sigmas(self, model, steps, denoise):
start_step = 10 - int(10 * denoise)
timesteps = torch.flip(torch.arange(1, 11) * 100 - 1, (0,))[start_step:start_step + steps]
ldm_patched.modules.model_management.load_models_gpu([model])
sigmas = model.model.model_sampling.sigma(timesteps)
sigmas = model.model_sampling.sigma(timesteps)
sigmas = torch.cat([sigmas, sigmas.new_zeros([1])])
return (sigmas, )