remove bloat from build-lite.py

NevermindNilas · Aug 17, 2024 · c40521f · c40521f
1 parent 17f2cbf
commit c40521f
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 8 deletions.
diff --git a/build-lite.py b/build-lite.py
@@ -55,12 +55,6 @@ def create_executable():
             "--hidden-import",
             "upscale_ncnn_py.upscale_ncnn_py_wrapper",
             "--collect-all",
-            "tensorrt",
-            "--collect-all",
-            "tensorrt-cu12-bindings",
-            "--collect-all",
-            "tensorrt_libs",
-            "--collect-all",
             "fastrlock",
             "--collect-all",
             "inquirer",

diff --git a/src/unifiedUpscale.py b/src/unifiedUpscale.py
@@ -407,14 +407,12 @@ def handleModel(self):
             (1, 3, self.height, self.width),
             device=self.deviceType,
             dtype=self.torchDType,
-            pin_memory=True,
         ).contiguous()
 
         self.dummyOutput = torch.zeros(
             (1, 3, self.height * self.upscaleFactor, self.width * self.upscaleFactor),
             device=self.deviceType,
             dtype=self.torchDType,
-            pin_memory=True,
         ).contiguous()
 
         self.IoBinding.bind_output(