Tiny improvements to endpoint_model.py, base_model.py,...

[sadra-barikbin]

Hi there!

This PR introduces some small improvements to models.

[NathanHB]

Hi ! Can you run the pre-commits hook on this ? It looks great we just need to make sure the CI passes.

pip install pre-commit
pre-commit install
pre-commit run --all-files

[clefourrier]

Hi! Do you want to fix the merge commits or drop the PR?

[HuggingFaceDocBuilderDev]

The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update.

[clefourrier]

LGTM

[clefourrier]

You'll need to fix conflicts and we'll be good to go :)