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

global unload after env var change

Browse Source
This commit is contained in:
layerdiffusion
2024-08-08 22:34:55 -07:00
parent 3f3cb12f76
commit 593455c4de
2 changed files with 13 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/main_entry.py
View File

@@ -1,7 +1,7 @@
import torch
import gradio as gr
from modules import shared_items, shared, ui_common, sd_models
from modules import shared_items, shared, ui_common, sd_models, processing
from modules import sd_vae as sd_vae_module
from backend import memory_management, stream
@@ -94,6 +94,8 @@ def refresh_memory_management_settings(model_memory, async_loading, pin_shared_m
print(f'Stream Used by CUDA: {stream.should_use_stream()}')
print(f'Current Inference Memory: {memory_management.minimum_inference_memory() / (1024 * 1024):.2f} MB')
print(f'PIN Shared Memory: {pin_shared_memory}')
processing.need_global_unload = True
return