File size: 466 Bytes
dd661b2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 |
# 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() |