batch_utils/__init__.py
| 1 | """Hybrid batched inference helpers for NVIDIA LocateAnything-3B.""" |
| 2 | from .hybrid_runtime import load, load_pil |
| 3 | from .engine_hybrid import generate_batch_hybrid, generate_batch_grouped_hybrid, get_last_hybrid_stats |
| 4 | |
| 5 | __all__ = [ |
| 6 | "load", |
| 7 | "load_pil", |
| 8 | "generate_batch_hybrid", |
| 9 | "generate_batch_grouped_hybrid", |
| 10 | "get_last_hybrid_stats", |
| 11 | ] |
| 12 | __version__ = "0.1.0" |
| 13 | |