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

Disable pin page

Browse Source 
This is an emergency fix

GTX 1060/1050/1066 either does not have shared GPU page vram or have less than 2GB shared page vram - pinning any tensors larger than that will crash

Solution is still under investigation.
This commit is contained in:
lllyasviel
2024-02-23 16:43:08 -08:00
parent 16caff3d14
commit 54c89503eb
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
ldm_patched/modules/model_management.py
View File

@@ -328,8 +328,8 @@ class LoadedModel:
else:
real_async_memory += module_mem
m.to(self.model.offload_device)
if is_device_cpu(self.model.offload_device):
m._apply(lambda x: x.pin_memory())
# if is_device_cpu(self.model.offload_device):
# m._apply(lambda x: x.pin_memory())
elif hasattr(m, "weight"):
m.to(self.device)
mem_counter += module_size(m)