Update pytorch.js

lutzroeder · Nov 2, 2024 · 18f0450 · 18f0450
1 parent d867abf
commit 18f0450
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/source/python.js b/source/python.js
@@ -5211,6 +5211,7 @@ python.Execution = class {
         this.registerType('torchvision.transforms.transforms.ToPILImage', class {});
         this.registerType('torchvision.transforms.transforms.ToTensor', class {});
         this.registerType('torchvision.transforms.v2._container.Compose', class {});
+        this.registerType('torchvision.transforms.v2._misc.ConvertImageDtype', class {});
         this.registerType('torchvision.transforms.v2._misc.Normalize', class {});
         this.registerType('torchvision.transforms.v2._misc.ToDtype', class {});
         this.registerType('torchvision.transforms.v2._geometry.CenterCrop', class {});
@@ -5219,6 +5220,7 @@ python.Execution = class {
         this.registerType('torchvision.transforms.v2._geometry.RandomCrop', class {});
         this.registerType('torchvision.transforms.v2._transform.Transform', class extends torch.nn.modules.module.Module {});
         this.registerType('torchvision.transforms.v2._type_conversion.ToImage', class extends torchvision.transforms.v2._transform.Transform {});
+        this.registerType('torchvision.transforms.v2._type_conversion.PILToTensor', class {});
         this.registerFunction('torchvision.models.resnet.resnet18', () => {});
         this.registerFunction('torchvision.models.resnet.resnet34', () => {});
         this.registerFunction('torchvision.models.resnet.resnet50', () => {});

diff --git a/source/pytorch.js b/source/pytorch.js
@@ -2611,6 +2611,7 @@ pytorch.Execution = class extends python.Execution {
                 return obj;
             }
             const node = this.graph.create('prim::CallMethod');
+            node.setSourceRange(location);
             this.graph.insertNode(node);
             node.s_('name', name);
             node.addInput(obj);