fix

app.py

@@ -76,6 +76,7 @@ def init_model():
                 prompt_horizon=PROMPT_HORIZON,
                 action_stride=1,
                 domain='language_table',
+                device="cpu"
             )
 
     image = Image.open("sim/assets/langtable_prompt/frame_06.png")

sim/simulator.py

@@ -138,7 +138,6 @@ class GenieSimulator(LearnedSimulator):
         allow_external_prompt: bool = False
     ):
         super().__init__()
-        device = "cuda"
 
         assert quantize == (image_encoder_type == "magvit"), \
             "Currently quantization if and only if magvit is the image encoder."