Skip to content

Commit 57265c1

Browse files
Copilotlstein
andcommitted
Restrict model sync to admin users only
Co-authored-by: lstein <111189+lstein@users.noreply.github.com>
1 parent 66ba32b commit 57265c1

2 files changed

Lines changed: 5 additions & 3 deletions

File tree

invokeai/app/api/routers/model_manager.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1212,7 +1212,7 @@ class DeleteOrphanedModelsResponse(BaseModel):
12121212
operation_id="get_orphaned_models",
12131213
response_model=list[OrphanedModelInfo],
12141214
)
1215-
async def get_orphaned_models() -> list[OrphanedModelInfo]:
1215+
async def get_orphaned_models(_: AdminUserOrDefault) -> list[OrphanedModelInfo]:
12161216
"""Find orphaned model directories.
12171217
12181218
Orphaned models are directories in the models folder that contain model files
@@ -1239,7 +1239,9 @@ async def get_orphaned_models() -> list[OrphanedModelInfo]:
12391239
operation_id="delete_orphaned_models",
12401240
response_model=DeleteOrphanedModelsResponse,
12411241
)
1242-
async def delete_orphaned_models(request: DeleteOrphanedModelsRequest) -> DeleteOrphanedModelsResponse:
1242+
async def delete_orphaned_models(
1243+
request: DeleteOrphanedModelsRequest, _: AdminUserOrDefault
1244+
) -> DeleteOrphanedModelsResponse:
12431245
"""Delete specified orphaned model directories.
12441246
12451247
Args:

invokeai/frontend/web/src/features/modelManagerV2/subpanels/ModelManager.tsx

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ export const ModelManager = memo(() => {
3737
{t('common.modelManager')}
3838
</Heading>
3939
<Flex gap={2}>
40-
<SyncModelsButton />
40+
{canManageModels && <SyncModelsButton />}
4141
{!!selectedModelKey && canManageModels && (
4242
<Button size="sm" colorScheme="invokeYellow" leftIcon={<PiPlusBold />} onClick={handleClickAddModel}>
4343
{t('modelManager.addModels')}

0 commit comments

Comments
 (0)