| # setup.py | |
| from huggingface_hub import snapshot_download | |
| def pre_setup(): | |
| print("🔧 Baixando modelo base para evitar erros de rede...") | |
| try: | |
| snapshot_download(repo_id="runwayml/stable-diffusion-v1-5", allow_patterns=["scheduler/*", "text_encoder/*", "tokenizer/*", "unet/*", "vae/*"]) | |
| print("✅ Modelo pré-baixado") | |
| except Exception as e: | |
| print(f"⚠️ Falha ao pré-baixar: {e}") | |
| if __name__ == "__main__": | |
| pre_setup() |