Spaces:
Running
on
L40S
Running
on
L40S
update
Browse files- infer_api.py +0 -2
infer_api.py
CHANGED
|
@@ -763,7 +763,6 @@ class InferSlrmAPI:
|
|
| 763 |
|
| 764 |
|
| 765 |
class InferMultiviewAPI:
|
| 766 |
-
@spaces.GPU
|
| 767 |
def __init__(self, config):
|
| 768 |
parser = argparse.ArgumentParser()
|
| 769 |
parser.add_argument("--seed", type=int, default=42)
|
|
@@ -815,7 +814,6 @@ class InferMultiviewAPI:
|
|
| 815 |
|
| 816 |
|
| 817 |
class InferCanonicalAPI:
|
| 818 |
-
@spaces.GPU
|
| 819 |
def __init__(self, config):
|
| 820 |
self.config = config
|
| 821 |
self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|
|
|
|
| 763 |
|
| 764 |
|
| 765 |
class InferMultiviewAPI:
|
|
|
|
| 766 |
def __init__(self, config):
|
| 767 |
parser = argparse.ArgumentParser()
|
| 768 |
parser.add_argument("--seed", type=int, default=42)
|
|
|
|
| 814 |
|
| 815 |
|
| 816 |
class InferCanonicalAPI:
|
|
|
|
| 817 |
def __init__(self, config):
|
| 818 |
self.config = config
|
| 819 |
self.device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
|