compressai_vision.model_wrappers#