xizaoqu
commited on
Commit
·
faeb2a7
1
Parent(s):
4caec38
update README
Browse files
app.py
CHANGED
|
@@ -162,6 +162,7 @@ class InteractiveRunner:
|
|
| 162 |
self.algo = algo
|
| 163 |
|
| 164 |
@spaces.GPU()
|
|
|
|
| 165 |
def run(self, first_frame, action, first_pose, curr_frame, device):
|
| 166 |
return self.algo.interactive(first_frame, action, first_pose, curr_frame, device=device)
|
| 167 |
|
|
@@ -252,7 +253,7 @@ def run(cfg: DictConfig):
|
|
| 252 |
global input_history
|
| 253 |
global memory_frames
|
| 254 |
|
| 255 |
-
runner.algo.to(device)
|
| 256 |
algodevice = next(runner.algo.parameters()).device
|
| 257 |
print(algodevice)
|
| 258 |
runner.algo.reset()
|
|
|
|
| 162 |
self.algo = algo
|
| 163 |
|
| 164 |
@spaces.GPU()
|
| 165 |
+
@torch.autocast("cuda")
|
| 166 |
def run(self, first_frame, action, first_pose, curr_frame, device):
|
| 167 |
return self.algo.interactive(first_frame, action, first_pose, curr_frame, device=device)
|
| 168 |
|
|
|
|
| 253 |
global input_history
|
| 254 |
global memory_frames
|
| 255 |
|
| 256 |
+
# runner.algo.to(device)
|
| 257 |
algodevice = next(runner.algo.parameters()).device
|
| 258 |
print(algodevice)
|
| 259 |
runner.algo.reset()
|