Do not reload worker when models change

bugbakery · Dec 17, 2023 · bab12c8 · bab12c8
1 parent 3418d87
commit bab12c8
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/worker/run.py b/worker/run.py
@@ -57,7 +57,11 @@ def main():
         stop_watching_event = asyncio.Event()
         p = run_sync_in_process(stop_watching_event, args)
 
-        for _ in watch(path, stop_event=stop_watching_event):
+        def watch_filter(change, path: str):
+            # do not reload when models are downloaded
+            return not path.startswith(settings.MODELS_DIR.absolute().as_posix())
+
+        for _ in watch(path, watch_filter=watch_filter, stop_event=stop_watching_event):
             logging.info("Source code change detected, reloading worker")
             p.terminate()
             p.join()