Quellcode durchsuchen

Update pytorch.js (#938)

Lutz Roeder vor 3 Jahren
Ursprung
Commit
df7b76ce63
1 geänderte Dateien mit 14 neuen und 9 gelöschten Zeilen
  1. 14 9
      source/pytorch.js

+ 14 - 9
source/pytorch.js

@@ -2520,13 +2520,16 @@ pytorch.Container.Zip = class {
             // https://github.com/pytorch/pytorch/blob/master/caffe2/serialize/inline_container.h
             // kProducedFileFormatVersion
             const versions = new Map([
-                [ '1', 'v1.3'  ],
-                [ '2', 'v1.5'  ], // 7a2889b014ce36fcc333b2c6de6f29f976652f84 (#28122)
-                [ '3', 'v1.6'  ], // 2ec6a30722b0ef85632a2f3e7ce6f80da403008a (#36085)
-                [ '4', 'v1.6'  ], // 95489b590f00801bdee7f41783f30874883cf6bb (#38620)
-                [ '5', 'v1.7'  ], // cb26661fe4faf26386703180a9045e6ac6d157df (#40364)
-                [ '6', 'v1.9'  ], // 3ee7637ffa50df0d9b231c7b40778ac1c390bf4a (#59714)
-                [ '7', 'v1.10' ]  // 880098a7e34a20628f960daa8eab0eb1ad566c39 (#63651)
+                [ '1',  'v1.3'  ],
+                [ '2',  'v1.5'  ], // 7a2889b014ce36fcc333b2c6de6f29f976652f84 (#28122)
+                [ '3',  'v1.6'  ], // 2ec6a30722b0ef85632a2f3e7ce6f80da403008a (#36085)
+                [ '4',  'v1.6'  ], // 95489b590f00801bdee7f41783f30874883cf6bb (#38620)
+                [ '5',  'v1.7'  ], // cb26661fe4faf26386703180a9045e6ac6d157df (#40364)
+                [ '6',  'v1.9'  ], // 3ee7637ffa50df0d9b231c7b40778ac1c390bf4a (#59714)
+                [ '7',  'v1.10' ], // 880098a7e34a20628f960daa8eab0eb1ad566c39 (#63651)
+                [ '8',  'v1.11' ], // b28e696516a7f0c7a6ead6da967590ce6c1d6698 (#71486)
+                [ '9',  'v1.11' ], // 8757e21c6a4fc00e83539aa7f9c28eb11eff53c1 (#72051)
+                [ '10', 'v1.12' ]  // 4f8b986e28736b59bc46cd0873a0f36fdaa6f5b8 (#61439)
             ]);
             if (!versions.has(value)) {
                 this._exceptionCallback(new pytorch.Error("Unsupported PyTorch Zip version '" + value + "'."));
@@ -2945,7 +2948,8 @@ pytorch.Container.Zip.Pickle = class extends pytorch.Container.Zip {
     }
 
     get format() {
-        return (this._entries.get('constants.pkl') ? 'TorchScript' : 'PyTorch') + ' ' + this.version('version');
+        const version = this.version('version') || this.version('.data/version');
+        return (this._entries.get('constants.pkl') ? 'TorchScript' : 'PyTorch') + (version ? ' ' + version : '');
     }
 
     get graphs() {
@@ -2981,7 +2985,8 @@ pytorch.Container.Zip.Package = class extends pytorch.Container.Zip {
     }
 
     get format() {
-        return 'PyTorch Package' + ' ' + this.version('.data/version');
+        const version = this.version('.data/version');
+        return 'PyTorch Package' + (version ? ' ' + version : '');
     }
 
     get graphs() {