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

set CUDA_VISIBLE_DEVICES with args

Browse Source
This commit is contained in:
lllyasviel
2024-02-05 23:52:50 -08:00
parent 1ecbff15fa
commit 74ff4a9ba9
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
modules_forge/initialization.py
View File

@@ -39,6 +39,10 @@ def initialize_forge():
args_parser.args, _ = args_parser.parser.parse_known_args()
if args_parser.args.gpu_device_id is not None:
os.environ['CUDA_VISIBLE_DEVICES'] = str(args_parser.args.gpu_device_id)
print("Set device to:", args_parser.args.gpu_device_id)
import ldm_patched.modules.model_management as model_management
import torch