Skip to content

Commit 4e86682

Browse files
committed
fix black screens with .clone()
1 parent 299ce49 commit 4e86682

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

backend/src/pytorch/InterpolateGIMM.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -175,7 +175,7 @@ def __call__(
175175

176176
output = output[:, :, : self.height, : self.width]
177177
retFrame = Frame(self.backend, self.width, self.height, img1.device, gpu_id=img1.gpu_id, hdr_mode=self.hdr_mode, dtype=img1.dtype)
178-
retFrame.set_frame_tensor(output)
178+
retFrame.set_frame_tensor(output.clone())
179179
yield retFrame
180180

181181
else:

backend/src/pytorch/InterpolateGMFSS.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -176,7 +176,7 @@ def __call__(
176176

177177
output = output[:, :, : self.height, : self.width]
178178
retFrame = Frame(self.backend, self.width, self.height, img1.device, gpu_id=img1.gpu_id, hdr_mode=self.hdr_mode, dtype=img1.dtype)
179-
retFrame.set_frame_tensor(output)
179+
retFrame.set_frame_tensor(output.clone())
180180
yield retFrame
181181
else:
182182
self.flownet.reset_cache_after_transition()

backend/src/pytorch/InterpolateIFRNET.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -149,7 +149,7 @@ def __call__(
149149
)
150150
for frame in frames:
151151
retFrame = Frame(self.backend, self.width, self.height, img1.device, gpu_id=img1.gpu_id, hdr_mode=self.hdr_mode, dtype=img1.dtype)
152-
retFrame.set_frame_tensor(frame[:, :self.height, :self.width])
152+
retFrame.set_frame_tensor(frame[:, :self.height, :self.width].clone())
153153
if not transition:
154154
yield retFrame
155155
self.torchUtils.copy_tensor(self.frame0, frame1, self.prepareStream)

backend/src/pytorch/InterpolateRIFE.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -487,7 +487,7 @@ def __call__(
487487
)
488488
output = output[:, :, : self.height, : self.width]
489489
retFrame = Frame(self.backend, self.width, self.height, img1.device, gpu_id=img1.gpu_id, hdr_mode=self.hdr_mode, dtype=img1.dtype)
490-
retFrame.set_frame_tensor(output)
490+
retFrame.set_frame_tensor(output.clone())
491491
yield retFrame
492492

493493
else:

backend/src/pytorch/UpscaleModelWrapper.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -105,4 +105,4 @@ def load_model(self, model=None) -> torch.nn.Module:
105105
self.inference_helper = TemporalSPANInferenceHelper(model=model, scale=self.__scale)
106106
def __call__(self, *args, **kwargs):
107107
assert self.inference_helper is not None, "Inference helper is not initialized."
108-
return self.inference_helper(*args, **kwargs)
108+
return self.inference_helper(*args, **kwargs).clone()

0 commit comments

Comments
 (0)