Skip to content

[tests] Refactor VLM detection in studio#71

Open
danielhanchen wants to merge 7 commits into
mainfrom
pr-5245-tests
Open

[tests] Refactor VLM detection in studio#71
danielhanchen wants to merge 7 commits into
mainfrom
pr-5245-tests

Conversation

@danielhanchen

Copy link
Copy Markdown
Member

Staging mirror of unslothai#5245

Original PR: unslothai#5245
Author: Datta0

This is a staging copy for review and editing. Once finalized, changes will be pushed back to the original PR.


Original description

Fixes : unslothai#5239. This seems to have not been an issue.

  • Add gemma4 as VLM to studio models list
  • Refactor is VLM checks to single place
  • Add a fallback to check vision config for the same.

This PR contains test changes only (1 files). Code changes are in the head PR.

Test files:

  • studio/backend/tests/test_vision_cache.py

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Studio doesn't detect gemma4-E4B as vision model

2 participants