Spaces:
Sleeping
Sleeping
update
Browse files- infer_api.py +0 -2
infer_api.py
CHANGED
|
@@ -811,7 +811,6 @@ class InferMultiviewAPI:
|
|
| 811 |
|
| 812 |
|
| 813 |
class InferCanonicalAPI:
|
| 814 |
-
@spaces.GPU
|
| 815 |
def __init__(self, config):
|
| 816 |
self.config = config
|
| 817 |
self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
@@ -821,7 +820,6 @@ class InferCanonicalAPI:
|
|
| 821 |
|
| 822 |
self.setup(**self.loaded_config)
|
| 823 |
|
| 824 |
-
@spaces.GPU
|
| 825 |
def setup(self,
|
| 826 |
validation: Dict,
|
| 827 |
pretrained_model_path: str,
|
|
|
|
| 811 |
|
| 812 |
|
| 813 |
class InferCanonicalAPI:
|
|
|
|
| 814 |
def __init__(self, config):
|
| 815 |
self.config = config
|
| 816 |
self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 820 |
|
| 821 |
self.setup(**self.loaded_config)
|
| 822 |
|
|
|
|
| 823 |
def setup(self,
|
| 824 |
validation: Dict,
|
| 825 |
pretrained_model_path: str,
|