Add abliterated (uncensored) text encoder support

- config.py: add UNCENSORED_TE_PATH pointing to bucket-mounted model
- app.py: load abliterated Qwen3 weights into pipe.text_encoder after
pipeline init, with graceful fallback to stock encoder
- Remove stale loras/ directory (superseded by bucket catalog)

Co-Authored-By: Oz <oz-agent@warp.dev>

README.md

@@ -8,5 +8,5 @@ sdk_version: 6.15.2
 app_file: app.py
 pinned: true
 license: apache-2.0
-short_description: try flux.2 klein9b loras, ablit
+short_description: try flux.2 klein9b loras + abliterated text encoder
 ---

app.py

@@ -19,6 +19,7 @@ from config import (
     MAX_SEED,
     MAX_LORA_SLOTS,
     PERSISTENT_LORA_CATALOG_PATH,
+    UNCENSORED_TE_PATH,
 )
 from ui_theme import orange_red_theme
 from upscale import UPSCALE_MODELS, apply_realesrgan
@@ -324,6 +325,17 @@ print(f"Loading FLUX.2 Klein {MODEL_VARIANT} from {MODEL_REPO}...")
 pipe = _PipeClass.from_pretrained(MODEL_REPO, torch_dtype=torch.bfloat16).to(device)
 print(f"Model loaded successfully: FLUX.2 Klein {MODEL_VARIANT}")
 
+# ── Replace text encoder with abliterated (uncensored) version ───────────
+if os.path.isfile(UNCENSORED_TE_PATH):
+    from safetensors.torch import load_file
+    print(f"Loading abliterated text encoder from {UNCENSORED_TE_PATH}...")
+    state_dict = load_file(UNCENSORED_TE_PATH)
+    pipe.text_encoder.load_state_dict(state_dict, strict=True)
+    pipe.text_encoder.to(device=device, dtype=torch.bfloat16)
+    print("Abliterated text encoder loaded — safety filters removed.")
+else:
+    print(f"Abliterated text encoder not found at {UNCENSORED_TE_PATH} — using stock encoder.")
+
 
 # ── UI helper callbacks ──────────────────────────────────────────────────────
 

config.py

@@ -29,3 +29,10 @@ PERSISTENT_LORA_CATALOG_PATH = os.environ.get(
     "PERSISTENT_LORA_CATALOG_PATH",
     "/loras-flux/config/lorasplayground.json",
 )
+
+# Abliterated (uncensored) text encoder — replaces the stock Qwen3 text encoder
+# weights after pipeline load. Set UNCENSORED_TE_PATH env var to override.
+UNCENSORED_TE_PATH = os.environ.get(
+    "UNCENSORED_TE_PATH",
+    "/loras-flux/fk9b-uncensored-text-encoder/model.safetensors",
+)