Spaces:

yslan
/

worldmem

Running on Zero

xizaoqu commited on Apr 8

Commit

4caec38

1 Parent(s): 0da4473

update README

Files changed (1) hide show

app.py CHANGED Viewed

@@ -252,6 +252,7 @@ def run(cfg: DictConfig):
         global input_history
         global memory_frames
         algodevice = next(runner.algo.parameters()).device
         print(algodevice)
         runner.algo.reset()

         global input_history
         global memory_frames
+        runner.algo.to(device)
         algodevice = next(runner.algo.parameters()).device
         print(algodevice)
         runner.algo.reset()