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()
|