diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md index 3bc02f6..34616c8 100644 --- a/docs/CHANGELOG.md +++ b/docs/CHANGELOG.md @@ -2,6 +2,13 @@ ## Unreleased changes +## 0.1.2 +- Added speed benchmarks for the UPET models +- Added a support for loading the local checkpoints +- Added variants selection for running different heads of the model +- Modifyed the internal TorchScript exporting mechanism, so it saves + a temporary model file on the disk. + ## 0.1.1 - Fixed a few bugs in the code, improved documentation and added a support for specifying the "latest" version of the model in the `upet.get_upet` and diff --git a/src/upet/__init__.py b/src/upet/__init__.py index 94f4490..6fb260d 100644 --- a/src/upet/__init__.py +++ b/src/upet/__init__.py @@ -1,4 +1,4 @@ -__version__ = "0.1.1" +__version__ = "0.1.2" from ._models import get_upet, save_upet