vcollos commited on
Commit
ed1cb43
·
verified ·
1 Parent(s): c1d02b3

Create hf_compat.py

Browse files
Files changed (1) hide show
  1. 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")