Skip to content

Commit 3100d98

Browse files
authored
Merge pull request #166 from TNTwise/2.3.7
fix pytorch not respecting gpu id
2 parents d03107d + b8acd13 commit 3100d98

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

backend/src/RenderVideo.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -392,6 +392,7 @@ def upscalePytorchObject(self, modelPath=None):
392392
height=self.height,
393393
backend=self.backend,
394394
tilesize=self.tilesize,
395+
gpu_id=self.pytorch_gpu_id,
395396
trt_optimization_level=self.trt_optimization_level,
396397
hdr_mode=self.hdr_mode,
397398
trt_static_shape= not self.trt_dynamic_shapes,
@@ -503,6 +504,7 @@ def setupInterpolate(self):
503504
device=self.device,
504505
dtype=self.precision,
505506
backend=self.backend,
507+
gpu_id=self.pytorch_gpu_id,
506508
UHDMode=self.UHD_mode,
507509
drba=self.drba,
508510
trt_optimization_level=self.trt_optimization_level,

0 commit comments

Comments
 (0)