add process destroy

huggingface · Nov 25, 2024 · 0f55511 · 0f55511
1 parent a71dabe
commit 0f55511
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/examples/inference/distributed/llava_next_video.py b/examples/inference/distributed/llava_next_video.py
@@ -185,6 +185,7 @@ def main(
             save_thread.shutdown(wait=True)
 
     save_future.result()
+    distributed_state.destroy_process_group()
 
 
 if __name__ == "__main__":