Update memory_management.py

memory_management.py

@@ -13,10 +13,6 @@ from pathlib import Path
 from enum import Enum
 from backend import stream, utils
 from backend.args import args
-gpu_sync = None
-if args.gpu_device_id is not None:
-    gpu_sync = GPUSync(args.gpu_device_id)
-
 
 cpu = torch.device('cpu')
 
@@ -143,6 +139,10 @@ class GPUSync:
         if Path("/tmp/sd_forge_gpu_sync.lock").exists():
             os.remove("/tmp/sd_forge_gpu_sync.lock")
 
+gpu_sync = None
+if args.gpu_device_id is not None:
+    gpu_sync = GPUSync(args.gpu_device_id)
+
 class VRAMState(Enum):
     DISABLED = 0  # No vram present: no need to move models to vram
     NO_VRAM = 1  # Very low vram: enable all the options to save vram