Alexander Soare a4891095e4

Use PytorchModelHubMixin to save models as safetensors (#125) ...

Co-authored-by: Remi <re.cadene@gmail.com>

2024-05-01 16:17:18 +01:00

..

act

Use PytorchModelHubMixin to save models as safetensors (#125)

2024-05-01 16:17:18 +01:00

diffusion

Use PytorchModelHubMixin to save models as safetensors (#125)

2024-05-01 16:17:18 +01:00

tdmpc

Tests cleaning & simplification (#81)

2024-04-18 14:47:42 +02:00

factory.py

Use PytorchModelHubMixin to save models as safetensors (#125)

2024-05-01 16:17:18 +01:00

normalize.py

Use PytorchModelHubMixin to save models as safetensors (#125)

2024-05-01 16:17:18 +01:00

policy_protocol.py

Use PytorchModelHubMixin to save models as safetensors (#125)

2024-05-01 16:17:18 +01:00

utils.py

ready for review

2024-04-12 11:36:52 +01:00