From c12d75c472932da476bf68b89254013a9e61aa6f Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Sun, 22 Feb 2026 19:56:12 +0000 Subject: [PATCH] Add transformers version switch to model config and vision check endpoints for dropdown selection --- studio/backend/routes/models.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/studio/backend/routes/models.py b/studio/backend/routes/models.py index 761c04d3e7..8747aefffa 100644 --- a/studio/backend/routes/models.py +++ b/studio/backend/routes/models.py @@ -248,6 +248,10 @@ async def get_model_config( This endpoint wraps the backend load_model_defaults function. """ try: + # Ensure correct transformers version for this model architecture + from utils.transformers_version import ensure_transformers_version + ensure_transformers_version(model_name) + logger.info(f"Getting model config for: {model_name}") # Load model defaults from backend config_dict = load_model_defaults(model_name) @@ -371,6 +375,10 @@ async def check_vision_model( This endpoint wraps the backend is_vision_model function. """ try: + # Ensure correct transformers version for this model architecture + from utils.transformers_version import ensure_transformers_version + ensure_transformers_version(model_name) + logger.info(f"Checking if vision model: {model_name}") is_vision = is_vision_model(model_name)