Update aduc_framework/managers/wan_manager.py

aduc_framework/managers/wan_manager.py

@@ -70,7 +70,7 @@ class WanManager:
         print("Loading models into memory. This may take a few minutes...")
 
         n_gpus = torch.cuda.device_count()
-        max_memory = {i: "45GiB" for i in range(n_gpus)}
+        max_memory = {i: "43GiB" for i in range(n_gpus)}
         max_memory["cpu"] = "120GiB"
 
         transformer = WanTransformer3DModel.from_pretrained(