File size: 1,362 Bytes
ed1cb43 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 |
"""
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") |