Allex21 commited on
Commit
dd661b2
·
verified ·
1 Parent(s): 3fdd31b

Create setup.py

Browse files
Files changed (1) hide show
  1. setup.py +13 -0
setup.py ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # setup.py
2
+ from huggingface_hub import snapshot_download
3
+
4
+ def pre_setup():
5
+ print("🔧 Baixando modelo base para evitar erros de rede...")
6
+ try:
7
+ snapshot_download(repo_id="runwayml/stable-diffusion-v1-5", allow_patterns=["scheduler/*", "text_encoder/*", "tokenizer/*", "unet/*", "vae/*"])
8
+ print("✅ Modelo pré-baixado")
9
+ except Exception as e:
10
+ print(f"⚠️ Falha ao pré-baixar: {e}")
11
+
12
+ if __name__ == "__main__":
13
+ pre_setup()