Wolfvin commited on
Commit
4147eb0
·
verified ·
1 Parent(s): e6557b8

Upload inference_example.py with huggingface_hub

Browse files
Files changed (1) hide show
  1. inference_example.py +3 -2
inference_example.py CHANGED
@@ -1,5 +1,5 @@
1
  #!/usr/bin/env python3
2
- """AAM Diffusion LLM v1.0 — Inference Example"""
3
 
4
  import sys
5
  from pathlib import Path
@@ -17,7 +17,7 @@ def main():
17
  # Create generator
18
  generator = AamGenerator(model, tokenizer, config)
19
 
20
- # Generate narrative from graph conditioning
21
  result = generator.generate(
22
  trigger="Siapa yang mencuri Snow Plum Pill?",
23
  evidence_nodes=["Hefei", "Diancang Five Swords", "Ju Jangmok"],
@@ -29,6 +29,7 @@ def main():
29
  print("=" * 60)
30
  print(" AAM Diffusion LLM — Generated Narrative")
31
  print("=" * 60)
 
32
  print(f" Narrative: {result.narrative}")
33
  print(f" Confidence: {result.confidence:.1%}")
34
  print(f" Steps: {result.n_diffusion_steps}")
 
1
  #!/usr/bin/env python3
2
+ """AAM Diffusion LLM — Inference Example"""
3
 
4
  import sys
5
  from pathlib import Path
 
17
  # Create generator
18
  generator = AamGenerator(model, tokenizer, config)
19
 
20
+ # Generate narrative
21
  result = generator.generate(
22
  trigger="Siapa yang mencuri Snow Plum Pill?",
23
  evidence_nodes=["Hefei", "Diancang Five Swords", "Ju Jangmok"],
 
29
  print("=" * 60)
30
  print(" AAM Diffusion LLM — Generated Narrative")
31
  print("=" * 60)
32
+ print(f" Trigger: {result.evidence_used}")
33
  print(f" Narrative: {result.narrative}")
34
  print(f" Confidence: {result.confidence:.1%}")
35
  print(f" Steps: {result.n_diffusion_steps}")