Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
fix: restore deprecated CompressionModel(entropy_bottleneck_channels)
Some external projects that use CompressAI as an importable library may depend on `CompressionModel` to create the `entropy_bottleneck`. Thus, we restore this legacy behavior (but discourage its use via a warning). ``` DeprecationWarning: The entropy_bottleneck_channels parameter is deprecated. Create an entropy_bottleneck in your model directly instead: class YourModel(CompressionModel): def __init__(self): super().__init__() self.entropy_bottleneck = EntropyBottleneck(entropy_bottleneck_channels) ```
- Loading branch information