lllyasviel/stable-diffusion-webui-forge
1
0
Fork 0
mirror of https://github.com/lllyasviel/stable-diffusion-webui-forge.git synced 2026-03-04 04:29:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

more api

Browse Source
This commit is contained in:
lllyasviel
2024-01-29 23:28:13 -08:00
parent 8ee804a159
commit a1d2a31233
4 changed files with 16 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
modules_forge/supported_preprocessor.py
View File

@@ -53,13 +53,7 @@ class Preprocessor:
def send_tensor_to_model_device(self, x):
return x.to(device=self.model_patcher.current_device, dtype=self.model_patcher.dtype)
def lazy_memory_management(self, model):
# This is a lazy method to just free some memory
# so that we can still use old codes to manage memory in a bad way
# Ideally this should all be removed and all memory should be managed by model patcher.
# But the workload is too big, so we just use a quick method to manage in dirty way.
required_memory = model_management.module_size(model) + model_management.minimum_inference_memory()
model_management.free_memory(required_memory, device=model_management.get_torch_device())
def process_after_running_preprocessors(self, process, params, *args, **kwargs):
return
def process_before_every_sampling(self, process, cond, *args, **kwargs):