File size: 1,309 Bytes
2d7e335 4147eb0 2d7e335 4147eb0 2d7e335 4147eb0 2d7e335 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | #!/usr/bin/env python3
"""AAM Diffusion LLM — Inference Example"""
import sys
from pathlib import Path
sys.path.insert(0, str(Path(__file__).parent))
import torch
from diffusion_llm import AamDiffusionModel, AamTokenizer, AamGenerator, AamDiffusionConfig
def main():
# Load model and tokenizer
config = AamDiffusionConfig.from_json("config.json")
model = AamDiffusionModel.load("model.pt", device="cpu")
tokenizer = AamTokenizer.load("tokenizer.json")
# Create generator
generator = AamGenerator(model, tokenizer, config)
# Generate narrative
result = generator.generate(
trigger="Siapa yang mencuri Snow Plum Pill?",
evidence_nodes=["Hefei", "Diancang Five Swords", "Ju Jangmok"],
anomalies=["Tidak ada konsumsi pil baru di pasar gelap"],
reasoning_steps=["Cross-reference tanggal kejadian", "Deteksi anomali pola"],
source_trust=0.85,
)
print("=" * 60)
print(" AAM Diffusion LLM — Generated Narrative")
print("=" * 60)
print(f" Trigger: {result.evidence_used}")
print(f" Narrative: {result.narrative}")
print(f" Confidence: {result.confidence:.1%}")
print(f" Steps: {result.n_diffusion_steps}")
print(f" Time: {result.generation_time_s:.2f}s")
if __name__ == "__main__":
main()
|