maybe speed up infernece?

TNTwise · May 15, 2024 · a520ed4 · a520ed4
1 parent 2087cd2
commit a520ed4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/torch/RenderCUDA.py b/src/torch/RenderCUDA.py
@@ -250,7 +250,7 @@ def FFmpegOut(self):
 
             #frame = np.ascontiguousarray(frame)
             self.main.imageDisplay = frame
-            self.writeProcess.stdin.buffer.write(frame.tobytes())
+            self.writeProcess.stdin.buffer.write(frame)
 
 
 class Interpolation(Render):