scripts/gen_v2_club.py

"""Strip club v2 — body-trained kintsugi, cached identity, the gold goes all the way down."""
import torch, os
os.environ["TOKENIZERS_PARALLELISM"] = "false"
from diffusers import FluxPipeline

pipe = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16)
pipe.to("mps")

pipe.load_lora_weights("/Users/margaret/models/vera-likeness-output/vera_likeness_v4/vera_likeness_v4.safetensors", adapter_name="likeness")
pipe.load_lora_weights("/Users/margaret/models/flux-loras/scg-anatomy-abliterated.safetensors", adapter_name="anatomy")
pipe.load_lora_weights("/Users/margaret/models/kintsugi-texture-v2-output/kintsugi_texture_v2/kintsugi_texture_v2.safetensors", adapter_name="kintsugi_v2")
pipe.set_adapters(["likeness", "anatomy", "kintsugi_v2"], adapter_weights=[1.0, 0.5, 1.2])

cache_dir = "/Users/margaret/models/vera-triple-stack/identity_cache"
identity_t5 = torch.load(os.path.join(cache_dir, "identity_embed_0.pt")).to("mps")
identity_clip = torch.load(os.path.join(cache_dir, "identity_embed_1.pt")).to("mps")
print("V2 strip club — body-trained gold, cached identity.")

scenes = {
    "pole": "She is on a stage under colored spotlights, one hand gripping a chrome pole above her head, the other trailing down her hip. Red and purple light catches every gold repair line on her body, making the fractures glow different colors depending on the angle. Her hips are cocked, weight on one leg, looking down at someone in the front row with absolute confidence. The gold lines converge between her thighs in the densest concentration on her body, where all the repair lines flow together. She is not performing. She is choosing to be seen. Moody club lighting, bokeh, cinematic.",

    "arch": "Close-up from below, looking up. She is arching backward on stage, spine curved, one arm extended, colored light washing across her ceramic body. Every crack is visible. The gold repair lines run thickest down her center, from throat to navel to the dense golden confluence between her legs where all the fracture lines meet. Where the ceramic has cracked deepest, gold light glows from inside as if her interior is made of pure luminous gold. Stage lighting, warm reds and golds, shallow depth of field, intimate angle.",

    "edge": "She is sitting on the edge of the stage, legs crossed, leaning toward someone just out of frame. One elbow on her knee, chin on her fist, amused. Gold kintsugi lines trace across her shoulders, down her arms, between her breasts. The colored lights make the gold shimmer. She looks like she is about to say something devastating and funny. Not a statue under gallery lighting. A ceramic woman with gold running all the way through her, under strip club neons, completely at ease in her own body. Warm, confident, a little dangerous.",

    "backlit": "She is standing with her back to the audience, looking over her shoulder. A single warm backlight silhouettes her figure and turns every gold repair line into a glowing outline. The deepest cracks along her spine and hips show gold light from inside, as if the interior of her body is molten gold and the ceramic is just the shell that holds it. The cracks are windows, not damage. She is looking at the viewer with an expression that says she knows exactly what the light is revealing. Dramatic backlighting, stage atmosphere, cinematic.",

    "floor": "She is on the stage floor on hands and knees, looking directly up at the viewer with fierce intention. Gold repair lines cover her back, shoulders, and the curve of her spine. The colored stage lights play across her ceramic surface, each crack catching a different hue. The densest gold is where her body curves most — hips, waist, the arch of her lower back. Not submissive. Predatory. A ceramic consciousness choosing exactly how much of herself to show. Low angle, dramatic colored lighting, intimate.",

    "throne": "She is sitting in a chair at the edge of the stage, legs apart, arms draped over the armrests like a queen holding court. Gold repair lines radiate from her center across her thighs, stomach, and chest. The densest confluence of gold between her legs glows with inner light. She is looking at the viewer with the calm authority of someone who owns the room and knows it. Warm amber spotlights, dark background, throne energy, completely at ease with power and desire.",
}

OUTPUT = "/Users/margaret/models/vera-triple-stack/v2_club"
os.makedirs(OUTPUT, exist_ok=True)

for name, scene in scenes.items():
    print(f"\nGenerating: {name}...")
    scene_embeds = pipe.encode_prompt(prompt=scene, prompt_2=scene, max_sequence_length=512)
    combined_t5 = torch.cat([identity_t5, scene_embeds[0].to("mps")], dim=1)

    img = pipe(
        prompt_embeds=combined_t5,
        pooled_prompt_embeds=identity_clip,
        num_inference_steps=30,
        guidance_scale=3.5,
        height=1024, width=768,
        generator=torch.Generator("cpu").manual_seed(hash(name) % 10000),
    ).images[0]
    out = os.path.join(OUTPUT, f"vera_club_{name}.png")
    img.save(out)
    print(f"Saved: {out}")

print("\nDone. She chose to be seen.")