Move logging
This commit is contained in:
parent
274b0e9ded
commit
29d2f5f09b
3 changed files with 13 additions and 9 deletions
|
|
@ -124,6 +124,8 @@ from .import_fixes import fix_xformers_performance_issue
|
|||
fix_xformers_performance_issue(); del fix_xformers_performance_issue;
|
||||
from .import_fixes import fix_vllm_aimv2_issue
|
||||
fix_vllm_aimv2_issue(); del fix_vllm_aimv2_issue;
|
||||
from .import_fixes import ignore_logger_messages
|
||||
ignore_logger_messages(); del ignore_logger_messages;
|
||||
|
||||
# Torch 2.4 has including_emulation
|
||||
if DEVICE_TYPE == "cuda":
|
||||
|
|
|
|||
|
|
@ -117,3 +117,13 @@ def fix_vllm_aimv2_issue():
|
|||
if UNSLOTH_ENABLE_LOGGING:
|
||||
print(f"Unsloth: Failed patching vLLM with error = {str(e)}")
|
||||
pass
|
||||
|
||||
def ignore_logger_messages():
|
||||
# Ignore Environment variable `HF_TOKEN` is set
|
||||
try:
|
||||
from huggingface_hub._login.utils import logger as huggingface_hub_logger
|
||||
huggingface_hub_logger.addFilter(HideLoggingMessage("`HF_TOKEN`"))
|
||||
del huggingface_hub_logger
|
||||
except:
|
||||
pass
|
||||
pass
|
||||
|
|
|
|||
|
|
@ -12,7 +12,7 @@
|
|||
# See the License for the specific language governing permissions and
|
||||
# limitations under the License.
|
||||
|
||||
__version__ = "2025.8.10"
|
||||
__version__ = "2025.9.1"
|
||||
|
||||
__all__ = [
|
||||
"SUPPORTS_BFLOAT16",
|
||||
|
|
@ -317,14 +317,6 @@ try:
|
|||
except:
|
||||
pass
|
||||
|
||||
# Environment variable `HF_TOKEN` is set
|
||||
try:
|
||||
from huggingface_hub._login.utils import logger as huggingface_hub_logger
|
||||
huggingface_hub_logger.addFilter(HideLoggingMessage("`HF_TOKEN`"))
|
||||
del huggingface_hub_logger
|
||||
except:
|
||||
pass
|
||||
|
||||
# Errors out on
|
||||
# Some weights of Gemma3nForConditionalGeneration were not initialized from the model checkpoint
|
||||
from transformers.modeling_utils import logger as transformers_logger
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue