diff --git a/src/main/java/io/bioimage/modelrunner/pytorch/JavaWorker.java b/src/main/java/io/bioimage/modelrunner/pytorch/JavaWorker.java
index 2de2304..eb10f35 100644
--- a/src/main/java/io/bioimage/modelrunner/pytorch/JavaWorker.java
+++ b/src/main/java/io/bioimage/modelrunner/pytorch/JavaWorker.java
@@ -29,6 +29,7 @@ public class JavaWorker {
 	 * 	args of the parent process
 	 */
 	public static void main(String[] args) {
+		System.clearProperty("PYTORCH_FLAVOR");
     	
     	try(Scanner scanner = new Scanner(System.in)){
     		PytorchInterface pi;