Skip to content

Auto model picker + FP16/FP32 precision selector (Upload UI + backend)#77

Open
eshankiyer wants to merge 3 commits into
BodyMaps:mainfrom
eshankiyer:eshankiyer-patch-2
Open

Auto model picker + FP16/FP32 precision selector (Upload UI + backend)#77
eshankiyer wants to merge 3 commits into
BodyMaps:mainfrom
eshankiyer:eshankiyer-patch-2

Conversation

@eshankiyer

Copy link
Copy Markdown

Adds automatic model selection and a precision control to the Upload pipeline, based on current main.

Frontend (UploadPage.tsx / .css): the model dropdown now defaults to an "Auto - AI picks the best model" option; on run it calls the backend to choose the model. A plain-language precision toggle (Faster / FP16 quantized vs Highest precision / FP32) and an optional free-text hint field were added. Precision is sent with every inference request.

Backend: run-epai-inference reads a precision field and threads it through _start_auto_segmentation -> run_auto_segmentation, which exports BODYMAPS_PRECISION to the GPU inference subprocess (ShapeKit, being CPU-only, is pinned to fp32). New /api/suggest-model endpoint uses the existing local Ollama client to pick a model from a small catalog, with a deterministic heuristic fallback when the LLM is unavailable.

No existing files are reverted; the diff is scoped to these changes.

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.

1 participant