Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

DirectML update.

Browse Source 
DirectML reuses GPU memory instead of returning it.
So prints "practical" GPU memory utilization too.
This commit is contained in:
Seunghoon Lee
2023-07-24 16:10:01 +09:00
parent 3da9642216
commit d4aa840a77
6 changed files with 68 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/shared.py
+2 -2
View File
@@ -12,7 +12,7 @@ import requests
import fasteners
from modules import errors, ui_components, shared_items, cmd_args
from modules.paths_internal import models_path, script_path, data_path, sd_configs_path, sd_default_config, sd_model_file, default_sd_model_file, extensions_dir, extensions_builtin_dir # pylint: disable=W0611
from modules.dml import directml_hijack_init, directml_override_opts
from modules.dml import directml_do_hijack, directml_override_opts
import modules.interrogate
import modules.memmon
import modules.styles
@@ -803,7 +803,7 @@ parallel_processing_allowed = not cmd_opts.lowvram
mem_mon = modules.memmon.MemUsageMonitor("MemMon", device, opts)
mem_mon.start()
if devices.backend == "directml":
directml_hijack_init()
directml_do_hijack()
directml_override_opts()