Skip to content

Commit 43b0d07

Browse files
feat(api): add route to reset hf token
1 parent f83592a commit 43b0d07

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

invokeai/app/api/routers/model_manager.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -893,6 +893,12 @@ def set_token(cls, token: str) -> HFTokenStatus:
893893
huggingface_hub.login(token=token, add_to_git_credential=False)
894894
return cls.get_status()
895895

896+
@classmethod
897+
def reset_token(cls) -> HFTokenStatus:
898+
with SuppressOutput(), contextlib.suppress(Exception):
899+
huggingface_hub.logout()
900+
return cls.get_status()
901+
896902

897903
@model_manager_router.get("/hf_login", operation_id="get_hf_login_status", response_model=HFTokenStatus)
898904
async def get_hf_login_status() -> HFTokenStatus:
@@ -915,3 +921,8 @@ async def do_hf_login(
915921
ApiDependencies.invoker.services.logger.warning("Unable to verify HF token")
916922

917923
return token_status
924+
925+
926+
@model_manager_router.delete("/hf_login", operation_id="reset_hf_token", response_model=HFTokenStatus)
927+
async def reset_hf_token() -> HFTokenStatus:
928+
return HFTokenHelper.reset_token()

0 commit comments

Comments
 (0)