From ebc5c3ddaafb9793862f3fbb291ddeefeb34683d Mon Sep 17 00:00:00 2001
From: Vladimir Mandic <mandic00@live.com>
Date: Thu, 13 Jun 2024 13:38:14 -0400
Subject: [PATCH] silence hf login

---
 modules/model_sd3.py | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/modules/model_sd3.py b/modules/model_sd3.py
index 085878dac..8eb9f8954 100644
--- a/modules/model_sd3.py
+++ b/modules/model_sd3.py
@@ -1,4 +1,6 @@
+import io
 import os
+import contextlib
 import warnings
 import torch
 import diffusers
@@ -16,8 +18,12 @@ def hf_login():
     import huggingface_hub as hf
     from modules import shared
     if shared.opts.huggingface_token is not None and len(shared.opts.huggingface_token) > 2 and not loggedin:
-        shared.log.debug(f'HF login token found: {"x" * len(shared.opts.huggingface_token)}')
-        hf.login(shared.opts.huggingface_token)
+        stdout = io.StringIO()
+        with contextlib.redirect_stdout(stdout):
+            hf.login(shared.opts.huggingface_token)
+        text = stdout.getvalue() or ''
+        line = [l for l in text.split('\n') if 'Token' in l]
+        shared.log.debug(f'HF login: {line[0] if len(line) > 0 else text}')
         loggedin = True