Spaces:

liruiw
/

hma

Running on Zero

liruiw commited on 19 days ago

Commit

b4e01e8

•

1 Parent(s): 4c4632b

tiny fix

Files changed (1) hide show

sim/simulator.py CHANGED Viewed

@@ -273,7 +273,7 @@ class GenieSimulator(LearnedSimulator):
             start_time = time.time()
         pred_next_latent_state = self.backbone.maskgit_generate(
             input_latent_states,
-            out_t=input_latent_states.shape[1] - 1,,
             maskgit_steps=self.inference_iterations,
             temperature=self.sampling_temperature,
             action_ids=input_actions,

             start_time = time.time()
         pred_next_latent_state = self.backbone.maskgit_generate(
             input_latent_states,
+            out_t=input_latent_states.shape[1] - 1,
             maskgit_steps=self.inference_iterations,
             temperature=self.sampling_temperature,
             action_ids=input_actions,