hf_compat.py

"""
Módulo de compatibilidade entre versões do huggingface_hub e diffusers.
Este arquivo cria uma ponte para a função cached_download ausente nas versões recentes.
"""

import os
import logging
from huggingface_hub import hf_hub_download
from huggingface_hub import HfFolder

logger = logging.getLogger(__name__)
logger.info("Carregando módulo de compatibilidade para huggingface_hub")

# Criando a função ausente como wrapper para a função atual
def cached_download(*args, **kwargs):
    """
    Função de compatibilidade que simula a função cached_download removida
    das versões recentes do huggingface_hub.
    """
    logger.info("Usando implementação compatível de cached_download")
    return hf_hub_download(*args, **kwargs)

# Adiciona a função ao módulo huggingface_hub
import huggingface_hub
if not hasattr(huggingface_hub, 'cached_download'):
    logger.info("Adicionando cached_download ao huggingface_hub")
    huggingface_hub.cached_download = cached_download

# Verifica se model_info também está ausente e fornece uma alternativa se necessário
if not hasattr(huggingface_hub, 'model_info'):
    logger.info("Adicionando model_info ao huggingface_hub")
    from huggingface_hub import model_info as _model_info
    huggingface_hub.model_info = _model_info

logger.info("Módulo de compatibilidade carregado com sucesso")