Skip to content

Conversation

@kaushikb11
Copy link
Contributor

@kaushikb11 kaushikb11 commented Feb 13, 2025

PR Description

Add automatic schema casting functionality to prediction responses across all prediction types
(image, document, video, audio). This allows responses to be automatically cast to either:

  • Domain-specific Pydantic models from the hub registry
  • Custom schemas provided via GenerationConfig

Key changes:

  • Add SchemaCastMixin for shared schema casting logic
  • Add vlmrun-hub as a core dependency

@kaushikb11 kaushikb11 merged commit ce77b1c into main Feb 14, 2025
4 checks passed
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.

2 participants