CoolFace
Modelpublic

spdev45/After-Earth-Director-8B

sourceHugging Facemitupdated 8mo agoView on Hugging Face
3likes77downloads
Model Card

After Earth: 2200 AC — Director-8B

The Technical Architect & Lore Engine for the 2200 AC Universe

After Earth: 2200 AC — Director-8B is a specialized large language model fine-tuned from Llama-3-8B to act as a Director in a multi-modal world-building pipeline. Its primary role is to translate abstract narrative or design intent into high-fidelity, technically grounded briefs suitable for downstream image generation models such as FLUX.1-Dev.

The model is optimized for hardware-efficient co-execution with diffusion models, enabling end-to-end generation on a single 24 GB VRAM GPU.

Generated Characters using the model

[image] [image] [image] [image] [image]


🚀 Model Capabilities

  • Technical Lore Generation Generates internally consistent, hard-sci-fi technical descriptions for the 2200 AC universe, including advanced materials such as Obsidian-Polymer and Aether-Conduits.
  • Asset Direction for Image Models Produces structured, optimized prompts designed for high-end diffusion models (e.g., FLUX.1-Dev).
  • Xeno-Biology & Non-Humanoid Design Capable of describing alien anatomy and non-human physiology beyond humanoid templates.
  • Hardware-Efficient Inference Optimized to run alongside diffusion pipelines on a single 24 GB GPU.

🛠️ Training Details

ComponentDescription
Base ArchitectureLlama-3-8B
Quantization4-bit (bitsandbytes)
Fine-TuningSupervised Fine-Tuning (SFT)
FrameworkUnsloth + Hugging Face TRL
Final Loss0.88
HardwareRTX 4090 (24 GB VRAM)
DatasetASTRA-SFT-1k + 2,000+ curated skins

📊 Technical Journey

Data Engineering & Ingestion

Secure ingestion and normalization of the After Earth: 2200 AC dataset.

The initial phase involved setting up a secure environment and ingesting the specialized After Earth: 2200 AC dataset.

Milestone: Successful authentication with Hugging Face CLI and mapping of the ASTRA-SFT-1k dataset.

Explanation: These screenshots document the setup of the persistent workspace and the ingestion of 2,000+ high-fidelity character skins. This data provides the visual foundation for modular armor, iridescence, and the 2200 AC "Aether-Tech" aesthetic. Snapshot of the output showing GPU (RTX 4090) and the "Unsloth is Ready!" message. [image] Built a Custom Pre-processing Pipeline, Performed schema validation to handle inconsistent dataset naming conventions [image] Curated a visual dataset of 2,000+ high-fidelity character skins. Filtered for 'Aether-Tech' aesthetics (Neural Cloud, Star Rail) to provide the visual foundation for 2200 AC bioluminescence and modular armor textures. Successfully implemented a targeted data-pull from the 'Aether-Tech' repository, resolving CLI authentication and repository-type mapping errors. [image] Training Log (the table showing Loss, Steps, and Learning Rate) [image] "Validation" image. It proves "Director" can now generate the high-quality technical prompts that will eventually drive image generator. [image]

Optimization & Convergence

Gradient checkpointing and CUDA cache management enabled stable scaling to 280 steps.

To ensure the 8B model could handle specialized lore without "forgetting" base logic, I implemented specific training constraints.

Milestone: Implementation of Gradient Checkpointing and manual CUDA cache clearing.

Explanation: These logs prove the ability to bypass VRAM bottlenecks on a single 24GB card. By monitoring these hyperparameters, I was able to scale training iterations from 100 to 280 steps without hardware failures.

Model has now completed over 2 full epochs (passing through entire dataset twice). The drop from 1.05 to 0.88 is significant. [image] Optimized VRAM utilization for extended training runs (280 steps) by implementing manual garbage collection and gradient checkpointing. Successfully managed a 24GB VRAM budget to prevent OOM (Out of Memory) crashes during the final convergence phase. [image] Monitored GPU thermals and memory frequency via nvidia-smi to ensure peak performance during a 280-step sustained training load. [image]

Training Convergence

This phase highlights the core fine-tuning work where the model was pushed to its final state.

Milestone: Final Training Loss of 0.88.

Explanation: These screenshots capture the live training logs. The drop in loss to 0.88 marks the transition from "Generalist Sci-Fi" to "Technical Lore Architect," where the model successfully learned to prioritize Obsidian-Polymer and Aether-Conduits in its outputs.

Final Training Loss: 0.8816 (Target < 1.0 Achieved) [image] Final validation of the 'Astra Nova Director' (280 steps). The model now exhibits high-density technical vocabulary and adheres strictly to the 2200 AC lore constraints without meta-commentary.

Demonstrating the transition from a conversational assistant to a specialized 'Lore Director' through iterative prompt engineering and low-temperature sampling. This ensures the output is ready for direct injection into a midjourney/stable-diffusion prompt pipeline. [image] Iteration 1: Logic Mapping.

[image] Iteration 2: Logic Mapping. [image]

Zero-Shot Validation

Validated portability on a fresh RunPod instance with autonomous technical inference.

Pulled the trained model and uploaded to the cloud back down to a fresh environment and made it work immediately. [image]


🖼️ Visual Asset Gallery: 2200 AC Concept Suite

Below are the production-ready assets generated by the Director-8B ➜ FLUX.1-Dev pipeline. Each asset demonstrates the model's ability to interpret complex material physics and non-humanoid biology.

🛡️ Guard & Combat Units

Unit ClassTechnical SpecificationsKey Visual Achievement
Monolith GuardQuadrupedal, Heavy Obsidian PlatingHard-Surface Geometry & Mass

[image] | Aether-Ghost | Translucent Skin, Floating Glass Shards | Sub-surface Scattering & Refraction | [image] | Hive-Shield | Chitinous Insectoid, Compound Eyes | Organic Texture Layering | [image]

🚀 Interstellar Vehicles

  • Aether-Interceptor A sleek starfighter featuring aerodynamic obsidian wings. Demonstrates the Director's capacity for industrial and mechanical design.

[image]


💻 Usage Example

python
from unsloth import FastLanguageModel
from diffusers import FluxPipeline
import torch, gc

model, tokenizer = FastLanguageModel.from_pretrained(
    "spdev45/After-Earth-Director-8B",
    load_in_4bit=True
)
FastLanguageModel.for_inference(model)

role = "High-ranking alien guardian in obsidian-polymer armor"
inputs = tokenizer(role, return_tensors="pt").to("cuda")
brief = tokenizer.decode(model.generate(**inputs, max_new_tokens=512)[0])

del model, tokenizer
gc.collect()
torch.cuda.empty_cache()

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

image = pipe(prompt=brief).images[0]
image.save("output.png")

👤 Author

Developed by spdev45


📄 License

MIT

  • Finetuned from model : unsloth/llama-3-8b-bnb-4bit

This llama model was trained 2x faster with Unsloth and Huggingface's TRL library.

<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>