Merge upstream PR 14855

modules_forge/forge_sampler.py

@@ -2,6 +2,7 @@ import torch
 from ldm_patched.modules.conds import CONDRegular, CONDCrossAttn
 from ldm_patched.modules.samplers import sampling_function
 from ldm_patched.modules import model_management
+from modules_forge.stream import synchronize_current_stream
 
 
 def cond_from_a1111_to_patched_ldm(cond):
@@ -113,4 +114,5 @@ def sampling_prepare(unet, x):
 def sampling_cleanup(unet):
     for cnet in unet.list_controlnets():
         cnet.cleanup()
+    synchronize_current_stream()
     return