| """ |
| Otimizador de carregamento de modelos para Hugging Face Spaces |
| """ |
| import os |
| import torch |
| from huggingface_hub import snapshot_download |
|
|
| class ModelOptimizer: |
| """Otimiza o carregamento de modelos para espaços com recursos limitados""" |
| |
| @staticmethod |
| def setup_models(): |
| """Configura modelos de forma otimizada""" |
| |
| device = "cuda" if torch.cuda.is_available() else "cpu" |
| |
| |
| if device == "cpu": |
| os.environ["OMP_NUM_THREADS"] = "1" |
| torch.set_num_threads(1) |
| |
| return device |
|
|
| |
| if __name__ == "__main__": |
| print("📥 Pré-carregando modelos para otimização...") |
| ModelOptimizer.setup_models() |
| print("✅ Modelos otimizados com sucesso!") |
