Spaces:
Runtime error
Runtime error
tiny fix
Browse files- sim/simulator.py +1 -1
sim/simulator.py
CHANGED
|
@@ -273,7 +273,7 @@ class GenieSimulator(LearnedSimulator):
|
|
| 273 |
start_time = time.time()
|
| 274 |
pred_next_latent_state = self.backbone.maskgit_generate(
|
| 275 |
input_latent_states,
|
| 276 |
-
out_t=input_latent_states.shape[1] - 1
|
| 277 |
maskgit_steps=self.inference_iterations,
|
| 278 |
temperature=self.sampling_temperature,
|
| 279 |
action_ids=input_actions,
|
|
|
|
| 273 |
start_time = time.time()
|
| 274 |
pred_next_latent_state = self.backbone.maskgit_generate(
|
| 275 |
input_latent_states,
|
| 276 |
+
out_t=input_latent_states.shape[1] - 1,
|
| 277 |
maskgit_steps=self.inference_iterations,
|
| 278 |
temperature=self.sampling_temperature,
|
| 279 |
action_ids=input_actions,
|