chore: ignore pyc and __pycache__

app.py

@@ -99,22 +99,46 @@ def get_pipe() -> FluxFillPipeline:
     global _PIPE
     if _PIPE is not None:
         return _PIPE
+
     device = "cuda" if torch.cuda.is_available() else "cpu"
     dtype = torch.bfloat16 if device == "cuda" else torch.float32
-    print(f"[pipe] load FLUX.1-Fill-dev dtype={dtype}, device={device}")
-    pipe = FluxFillPipeline.from_pretrained("black-forest-labs/FLUX.1-Fill-dev", torch_dtype=dtype).to(device)
-
-    # LoRA（stage1）
-    lora_dir = CODE_EDIT / "stage1" / "checkpoint-4800"
-    if lora_dir.exists():
-        try:
-            pipe.load_lora_weights(str(lora_dir))  # 需要 peft
-            print(f"[pipe] loaded LoRA from: {lora_dir}")
-        except Exception as e:
-            print(f"[pipe] load LoRA failed (continue without): {e}")
-    else:
-        print(f"[pipe] LoRA path not found: {lora_dir} (continue without)")
 
+
+    local_flux = BASE_DIR / "code_edit" / "flux_cache"  
+    use_local = local_flux.exists()
+
+
+    hf_token = os.environ.get("HF_TOKEN")
+
+
+    try:
+        from huggingface_hub import hf_hub_enable_hf_transfer
+        hf_hub_enable_hf_transfer()
+    except Exception:
+        pass
+
+    print(f"[pipe] loading FLUX.1-Fill-dev (dtype={dtype}, device={device}, local={use_local})")
+    try:
+        if use_local:
+            pipe = FluxFillPipeline.from_pretrained(
+                local_flux, torch_dtype=dtype
+            ).to(device)
+        else:
+            # 在线拉取（需要 gated 访问 + token）
+            pipe = FluxFillPipeline.from_pretrained(
+                "black-forest-labs/FLUX.1-Fill-dev",
+                torch_dtype=dtype,
+                token=hf_token,           
+                # use_auth_token=hf_token, 
+            ).to(device)
+    except Exception as e:
+        raise RuntimeError(
+            "Failed to load FLUX.1-Fill-dev. "
+            "Make sure your account has access to the gated repo and HF_TOKEN is set as a Space secret, "
+            "or pre-download to a local cache directory."
+        ) from e
+
+    # ……你原有的 LoRA 加载逻辑……
     _PIPE = pipe
     return pipe