xizaoqu commited on
Commit
faeb2a7
·
1 Parent(s): 4caec38

update README

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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()