Skip to content

Commit 68e1089

Browse files
Upgrade pytorch version
Signed-off-by: Nathalie Jonathan <[email protected]>
1 parent 31f0422 commit 68e1089

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/DLModel.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -253,7 +253,7 @@ protected void loadModel(
253253
ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
254254
try {
255255
System.setProperty("PYTORCH_PRECXX11", "true");
256-
System.setProperty("PYTORCH_VERSION", "1.13.1");
256+
System.setProperty("PYTORCH_VERSION", "2.5.1");
257257
System.setProperty("DJL_CACHE_DIR", mlEngine.getMlCachePath().toAbsolutePath().toString());
258258
// DJL will read "/usr/java/packages/lib" if don't set "java.library.path". That will throw
259259
// access denied exception

ml-algorithms/src/main/java/org/opensearch/ml/engine/algorithms/DLModelExecute.java

+1-1
Original file line numberDiff line numberDiff line change
@@ -131,7 +131,7 @@ private void loadModel(File modelZipFile, String modelId, String modelName, Stri
131131
ClassLoader contextClassLoader = Thread.currentThread().getContextClassLoader();
132132
try {
133133
System.setProperty("PYTORCH_PRECXX11", "true");
134-
System.setProperty("PYTORCH_VERSION", "1.13.1");
134+
System.setProperty("PYTORCH_VERSION", "2.5.1");
135135
System.setProperty("DJL_CACHE_DIR", mlEngine.getMlCachePath().toAbsolutePath().toString());
136136
// DJL will read "/usr/java/packages/lib" if don't set "java.library.path". That will throw
137137
// access denied exception

0 commit comments

Comments
 (0)