Skip to content

Commit

Permalink
add new model with more lag, reduce temp a bit
Browse files Browse the repository at this point in the history
  • Loading branch information
nuwandavek committed May 30, 2024
1 parent f6491cb commit dba6893
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 1 deletion.
Binary file modified models/tinyphysics.onnx
Binary file not shown.
2 changes: 1 addition & 1 deletion tinyphysics.py
Original file line number Diff line number Diff line change
Expand Up @@ -86,7 +86,7 @@ def get_current_lataccel(self, sim_states: List[State], actions: List[float], pa
'states': np.expand_dims(states, axis=0).astype(np.float32),
'tokens': np.expand_dims(tokenized_actions, axis=0).astype(np.int64)
}
return self.tokenizer.decode(self.predict(input_data, temperature=1.))
return self.tokenizer.decode(self.predict(input_data, temperature=0.8))


class TinyPhysicsSimulator:
Expand Down

0 comments on commit dba6893

Please sign in to comment.