This update replaces the earlier checkpoint and its shift-1 inference recipe.
Use 4 steps, CFG 1, scheduler shift 3, with raw sigmas [1, .75, .5, .25].
The scheduler applies the shift once, producing [1, .9, .75, .5, 0].
The included custom transformer defaults to K=4 conditioning. A plain
CosmosTransformer3DModel cannot load the full checkpoint correctly: it would drop
the learned K branch. Use the supplied module as shown below.
Local loading and generation
Requires PyTorch with CUDA/BF16, Diffusers with Anima modular support, Transformers, Accelerate and Safetensors. Validated environment: Diffusers 0.39.0, Transformers 5.7.0 and PyTorch 2.12.1+cu130.
Download the complete repository with hf download aina-tech/Anima-Lightning --local-dir Anima-Lightning.
Review the included Python module before importing it.
from pathlib import Path
import sys
import torch
from diffusers import AnimaModularPipeline
local = str(Path("Anima-Lightning").resolve())
sys.path.insert(0, local)
import anima_lightning_transformer
pipe = AnimaModularPipeline.from_pretrained(local, local_files_only=True)
pipe.load_components(
["text_encoder", "tokenizer", "t5_tokenizer", "text_conditioner",
"transformer", "scheduler", "vae"],
pretrained_model_name_or_path=local,
local_files_only=True, torch_dtype=torch.bfloat16,
)
pipe.guider.guidance_scale = 1.0
pipe.to("cuda")
image = pipe(
prompt="anime illustration, a lighthouse above a calm sea, sunset",
negative_prompt=None, width=1024, height=1024,
num_inference_steps=4, sigmas=[1.0, 0.75, 0.5, 0.25],
max_sequence_length=512,
generator=torch.Generator(device="cpu").manual_seed(42),
output="images",
)[0]
image.save("anima_lightning.png")
Do not use the former shift-1 recipe or standard full-step CFG5 inference.
The saved transformer tensors are BF16; results can differ from the FP32 checkpoint
and from other sampling implementations. step400_export.json records source hashes,
the complete tensor inventory, and the BF16 export validation.
Model and license
Derived from CircleStone Labs Anima and NVIDIA Cosmos Predict2. This is an independent
distilled release, intended for anime and illustration. Anatomy, complex compositions,
and text rendering can be imperfect. See the unchanged LICENSE.md and NOTICE.md.
The CircleStone Labs non-commercial license and applicable upstream terms remain in effect.