comfyanonymous/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-03-13 00:59:59 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(isolation-lifecycle): execution/model ejection parity + fenced sampler device handling

Browse Source 
add pyisolate==0.9.1 to requirements.txt
This commit is contained in:
John Pollock
2026-02-27 13:07:23 -06:00
parent a1c3124821
commit 3c8ba051b6
8 changed files with 229 additions and 56 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