zlei6/ComfyUI
1
0
Fork 0
mirror of https://github.com/Comfy-Org/ComfyUI.git synced 2026-08-24 18:10:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(isolation): sandbox policy and runtime fencing

Browse Source
This commit is contained in:
John Pollock
2026-04-30 14:51:29 -05:00
parent fbb6be5624
commit c9ebc6aa57
8 changed files with 388 additions and 34 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
cuda_malloc.py
View File

@@ -92,7 +92,7 @@ if args.cuda_malloc:
env_var = os.environ.get('PYTORCH_CUDA_ALLOC_CONF', None)
if env_var is None:
env_var = "backend:cudaMallocAsync"
else:
elif not args.use_process_isolation:
env_var += ",backend:cudaMallocAsync"
os.environ['PYTORCH_CUDA_ALLOC_CONF'] = env_var