refactor: pass model version in save_load #5930

Job	Run time
manylinux.amd64.py3.6.build	4m 15s
manylinux.amd64.py3.7.build	4m 6s
manylinux.amd64.py3.8.build	4m 10s
manylinux.amd64.py3.9.build	4m 8s
manylinux.amd64.py3.10.build	4m 7s
manylinux.aarch64.py${{ matrix.config.version }}.build	0s
ubuntu-latest.amd64.py3.8.sdist-bundle	43s
windows-2019.amd64.py3.6.build	14m 29s
windows-2019.amd64.py3.7.build	16m 9s
windows-2019.amd64.py3.8.build	16m 3s
windows-2019.amd64.py3.9.build	19m 46s
windows-2019.amd64.py3.10.build	17m 18s
macos.amd64.py3.6.build	10m 42s
macos.amd64.py3.7.build	9m 34s
macos.amd64.py3.8.build	10m 33s
macos.amd64.py3.9.build	10m 0s
macos.amd64.py3.10.build	16m 0s
manylinux.aarch64.py${{ matrix.config.version }}.test	0s
ubuntu-latest.amd64.py3.8.sdist-build-test	5m 17s
manylinux.amd64.py3.6.test	2m 47s
manylinux.amd64.py3.7.test	2m 47s
manylinux.amd64.py3.8.test	2m 5s
manylinux.amd64.py3.9.test	2m 21s
manylinux.amd64.py3.10.test	2m 18s
macos.amd64.py3.6.test	1m 16s
macos.amd64.py3.7.test	5m 0s
macos.amd64.py3.8.test	12m 33s
macos.amd64.py3.9.test	13m 29s
macos.amd64.py3.10.test	6h 0m 14s
windows-2019.amd64.py3.6.test	7m 21s
windows-2019.amd64.py3.7.test	7m 14s
windows-2019.amd64.py3.8.test	7m 3s
windows-2019.amd64.py3.9.test	7m 7s
windows-2019.amd64.py3.10.test	7m 1s
	10h 7m 56s

Provide feedback