Load .tar.gz on load_models=all (#3329)

pytorch · Sep 23, 2024 · 6881ec5 · 6881ec5
1 parent d993070
commit 6881ec5
Showing 1 changed file with 2 additions and 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
@@ -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 {