Commit Graph

13 Commits

Author SHA1 Message Date
Simon Alibert 75dd478b54 Revert "Revert "Override action queue defaults""
This reverts commit 5647d71c08.
2024-05-03 11:05:59 +02:00
Simon Alibert 5647d71c08 Revert "Override action queue defaults"
This reverts commit 7592c21eb2751c4ecc0fc1badb437f34b0f2155e.
2024-05-03 10:04:49 +02:00
Simon Alibert 61f38da7a3 Override action queue defaults 2024-05-03 10:04:49 +02:00
Simon Alibert d08ff9e9a0 rename make_optimizer 2024-05-03 10:04:49 +02:00
Simon Alibert 6f11b0afaf Cleanup 2024-05-03 10:04:49 +02:00
Simon Alibert e75448c8c4 Generate artifacts from x86 cpu 2024-05-03 10:04:49 +02:00
Simon Alibert 77ee567851 Add require_cpu 2024-05-03 10:04:49 +02:00
Simon Alibert 344c1653f2 Fix device 2024-05-03 10:04:49 +02:00
Simon Alibert 5877eb38ab Enable backward compatibility testing on tdmpc 2024-05-03 10:04:49 +02:00
Simon Alibert 3a918b980f WIP 2024-05-03 10:04:48 +02:00
Simon Alibert 7002813bbb Cleanup, add output_dict 2024-05-03 10:04:48 +02:00
Simon Alibert 2726d1a6c9 Add get_policy_stats 2024-05-03 10:04:48 +02:00
Simon Alibert 6e15e9664b WIP Add save_policy_to_safetensors 2024-05-03 10:04:48 +02:00