Skip to content

Commit 8b391bc

Browse files
committed
Load .tar.gz on load_models=all
1 parent ba8c268 commit 8b391bc

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

frontend/server/src/main/java/org/pytorch/serve/ModelServer.java

+2-1
Original file line numberDiff line numberDiff line change
@@ -193,7 +193,8 @@ private void initModelStore() throws InvalidSnapshotException, IOException {
193193
String fileName = file.getName();
194194
if (file.isFile()
195195
&& !fileName.endsWith(".mar")
196-
&& !fileName.endsWith(".model")) {
196+
&& !fileName.endsWith(".model")
197+
&& !fileName.endsWith(".tar.gz")) {
197198
continue;
198199
}
199200
try {

0 commit comments

Comments
 (0)