From 09e4e73f6e6bc80ec0ef41488c530eb54f2cac5e Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Thu, 22 Jun 2023 21:37:10 +0200
Subject: [PATCH] Adding new ignore_rule for conversion.

Hesitant to whitelist `pytorch_model` instead tbh.
---
 server/text_generation_server/utils/hub.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/server/text_generation_server/utils/hub.py b/server/text_generation_server/utils/hub.py
index fbb570a6..23743c9b 100644
--- a/server/text_generation_server/utils/hub.py
+++ b/server/text_generation_server/utils/hub.py
@@ -30,6 +30,7 @@ def weight_hub_files(
         and len(s.rfilename.split("/")) == 1
         and "arguments" not in s.rfilename
         and "args" not in s.rfilename
+        and "training" not in s.rfilename
     ]
 
     if not filenames: