From 8b391bc52de46b777d04c2ac34218df600023f7b Mon Sep 17 00:00:00 2001
From: m10an <vn.mlshk@gmail.com>
Date: Wed, 18 Sep 2024 19:07:36 +0200
Subject: [PATCH] Load .tar.gz on `load_models=all`

---
 .../server/src/main/java/org/pytorch/serve/ModelServer.java    | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/frontend/server/src/main/java/org/pytorch/serve/ModelServer.java b/frontend/server/src/main/java/org/pytorch/serve/ModelServer.java
index f5b02ee222..90f983333a 100644
--- a/frontend/server/src/main/java/org/pytorch/serve/ModelServer.java
+++ b/frontend/server/src/main/java/org/pytorch/serve/ModelServer.java
@@ -193,7 +193,8 @@ private void initModelStore() throws InvalidSnapshotException, IOException {
                     String fileName = file.getName();
                     if (file.isFile()
                             && !fileName.endsWith(".mar")
-                            && !fileName.endsWith(".model")) {
+                            && !fileName.endsWith(".model")
+                            && !fileName.endsWith(".tar.gz")) {
                         continue;
                     }
                     try {