mirror of
https://github.com/jamiepine/voicebox.git
synced 2026-10-03 17:15:19 -07:00
fix(models): honor model_name on /models/load and /models/unload (fixes #977)
This commit is contained in:
committed by
capy-ai-staging[bot]
parent
d09c5c39e3
commit
cf0da2e6a3
@@ -489,6 +489,19 @@ class ModelDownloadRequest(BaseModel):
|
||||
model_name: str
|
||||
|
||||
|
||||
class ModelLoadRequest(BaseModel):
|
||||
"""Request model for loading or unloading a model by name.
|
||||
|
||||
``model_name`` is one of the ids returned by ``GET /models/status``
|
||||
(e.g. ``"kokoro"``, ``"qwen-tts-0.6B"``, ``"whisper-turbo"``).
|
||||
``model_size`` is the legacy Qwen-only selector, kept so existing
|
||||
``POST /models/load?model_size=0.6B`` callers keep working.
|
||||
"""
|
||||
|
||||
model_name: Optional[str] = None
|
||||
model_size: Optional[str] = None
|
||||
|
||||
|
||||
class ModelMigrateRequest(BaseModel):
|
||||
"""Request model for migrating models to a new directory."""
|
||||
|
||||
|
||||
Reference in New Issue
Block a user