Create hf_compat.py
Browse files- hf_compat.py +35 -0
hf_compat.py
ADDED
@@ -0,0 +1,35 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
"""
|
2 |
+
Módulo de compatibilidade entre versões do huggingface_hub e diffusers.
|
3 |
+
Este arquivo cria uma ponte para a função cached_download ausente nas versões recentes.
|
4 |
+
"""
|
5 |
+
|
6 |
+
import os
|
7 |
+
import logging
|
8 |
+
from huggingface_hub import hf_hub_download
|
9 |
+
from huggingface_hub import HfFolder
|
10 |
+
|
11 |
+
logger = logging.getLogger(__name__)
|
12 |
+
logger.info("Carregando módulo de compatibilidade para huggingface_hub")
|
13 |
+
|
14 |
+
# Criando a função ausente como wrapper para a função atual
|
15 |
+
def cached_download(*args, **kwargs):
|
16 |
+
"""
|
17 |
+
Função de compatibilidade que simula a função cached_download removida
|
18 |
+
das versões recentes do huggingface_hub.
|
19 |
+
"""
|
20 |
+
logger.info("Usando implementação compatível de cached_download")
|
21 |
+
return hf_hub_download(*args, **kwargs)
|
22 |
+
|
23 |
+
# Adiciona a função ao módulo huggingface_hub
|
24 |
+
import huggingface_hub
|
25 |
+
if not hasattr(huggingface_hub, 'cached_download'):
|
26 |
+
logger.info("Adicionando cached_download ao huggingface_hub")
|
27 |
+
huggingface_hub.cached_download = cached_download
|
28 |
+
|
29 |
+
# Verifica se model_info também está ausente e fornece uma alternativa se necessário
|
30 |
+
if not hasattr(huggingface_hub, 'model_info'):
|
31 |
+
logger.info("Adicionando model_info ao huggingface_hub")
|
32 |
+
from huggingface_hub import model_info as _model_info
|
33 |
+
huggingface_hub.model_info = _model_info
|
34 |
+
|
35 |
+
logger.info("Módulo de compatibilidade carregado com sucesso")
|