Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
4c7d24d
1
Parent(s):
f6a1fc4
dkt/pipelines/pipeline.py
CHANGED
|
@@ -956,7 +956,7 @@ class DKTPipeline:
|
|
| 956 |
|
| 957 |
|
| 958 |
|
| 959 |
-
@spaces.GPU()
|
| 960 |
@torch.inference_mode()
|
| 961 |
def moge_infer(self, input_image):
|
| 962 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 956 |
|
| 957 |
|
| 958 |
|
| 959 |
+
@spaces.GPU(duration=30)
|
| 960 |
@torch.inference_mode()
|
| 961 |
def moge_infer(self, input_image):
|
| 962 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|