From 39450e47e9fdc29887fcc1db866a8dd66ae027c1 Mon Sep 17 00:00:00 2001
From: Michael Benayoun <mickbenayoun@gmail.com>
Date: Thu, 2 Nov 2023 10:51:00 +0100
Subject: [PATCH] Fix neuron cache starting compilation before fetching (#280)

Waiting on fetch
---
 optimum/neuron/trainers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/optimum/neuron/trainers.py b/optimum/neuron/trainers.py
index 72047d479..2de816818 100755
--- a/optimum/neuron/trainers.py
+++ b/optimum/neuron/trainers.py
@@ -182,7 +182,7 @@ def __init__(self, *args, **kwargs):
             original_neuron_cache_path=_ORIGINAL_NEURON_CACHE_PATH,
             fetch=fetch,
             push=push,
-            wait_for_everyone_on_fetch=False,
+            wait_for_everyone_on_fetch=True,
             wait_for_everyone_on_push=True,
         )
         self.add_callback(callback)