Spaces:

yslan
/

worldmem

Running on Zero

xizaoqu commited on Apr 8

Commit

0da4473

1 Parent(s): 1cccab7

update README

Files changed (1) hide show

app.py CHANGED Viewed

@@ -252,8 +252,8 @@ def run(cfg: DictConfig):
         global input_history
         global memory_frames
-        # device = next(runner.algo.parameters()).device
-        # print(device)
         runner.algo.reset()
         memory_frames = []
         memory_frames.append(load_image_as_tensor(DEFAULT_IMAGE))

         global input_history
         global memory_frames
+        algodevice = next(runner.algo.parameters()).device
+        print(algodevice)
         runner.algo.reset()
         memory_frames = []
         memory_frames.append(load_image_as_tensor(DEFAULT_IMAGE))