app

app.py

@@ -55,7 +55,8 @@ def init_pipeline_if_needed():
     global pipe
     if pipe is not None:
         return
-
+    
+    print("🚀 Initializing pipeline...")
     if use_int8 or get_gpu_memory_gb() < 33:
         transformer_model = FluxTransformer2DModel.from_pretrained(
             "sayakpaul/flux.1-schell-int8wo-improved",
@@ -76,10 +77,13 @@ def init_pipeline_if_needed():
             token=hf_token,
         )
 
+    print("📦 Loading model to CUDA...")
     _pipe = _pipe.to("cuda")
+    print("🎨 Loading LoRA weights...")
     _pipe.load_lora_weights(MODEL_PATH, adapter_name=ADAPTER_NAME)
     _pipe.set_adapters([ADAPTER_NAME])
     pipe = _pipe
+    print("✅ Pipeline initialized successfully!")
 
 def _to_pil_rgba(img: Any) -> Image.Image:
     """Convert various inputs to PIL RGBA image"""
@@ -454,6 +458,15 @@ def create_simple_app():
     
     return app
 
+# 在应用启动前预加载模型
+print("=" * 60)
+print("🔧 Pre-loading models on startup...")
+print("=" * 60)
+init_pipeline_if_needed()
+print("=" * 60)
+print("✨ All models loaded and ready!")
+print("=" * 60)
+
 if __name__ == "__main__":
     app = create_simple_app()
     app.launch(