Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
da65577
1
Parent(s):
3ba6a3b
dkt/pipelines/pipeline.py
CHANGED
|
@@ -993,11 +993,8 @@ class DKTPipeline:
|
|
| 993 |
input_image = input_image.unsqueeze(0) # (1, 3, H, W)
|
| 994 |
|
| 995 |
|
| 996 |
-
|
| 997 |
-
|
| 998 |
-
|
| 999 |
model_device = next(self.moge_pipe.parameters()).device
|
| 1000 |
-
print(f'input_image device: {input_image.device}, moge_pipe device: {model_device}')
|
| 1001 |
|
| 1002 |
return self.moge_pipe.infer(input_image)
|
| 1003 |
|
|
|
|
| 993 |
input_image = input_image.unsqueeze(0) # (1, 3, H, W)
|
| 994 |
|
| 995 |
|
|
|
|
|
|
|
|
|
|
| 996 |
model_device = next(self.moge_pipe.parameters()).device
|
| 997 |
+
print(f'input_image device: {input_image.device}, moge_pipe device: {model_device}, input shape: {input_image.shape}, input dtype: {input_image.dtype}')
|
| 998 |
|
| 999 |
return self.moge_pipe.infer(input_image)
|
| 1000 |
|