anandkaman's picture
SDK source: assets/sdk/ (8 files; pip install controlmt)
c14da2b verified
Raw
History Blame Contribute Delete
1.1 kB
"""ControlMT — Python SDK for the ControlMT v2.3 KN↔EN translator.
Drop-in entry point for users who don't want to wire HuggingFace Transformers
themselves. Picks the right device + dtype + quantization automatically;
overridable when you need control.
Quick start:
from controlmt import ControlMT
model = ControlMT.from_hf() # auto everything
print(model.translate("ನಾನು ಕನ್ನಡ ಮಾತನಾಡುತ್ತೇನೆ.")) # → "I speak Kannada."
Explicit:
model = ControlMT.from_hf(device="cpu", quant="int8") # int8 dynamic on CPU
model = ControlMT.from_hf(device="gpu", dtype="float16") # fp16 on GPU
model = ControlMT.from_hf(device="auto") # GPU-with-CPU-fallback (default)
Batched (user-defined size, defaults to 1 = no batching):
model.batch_translate(texts, batch_size=8)
model.batch_translate(texts, auto_batch=True) # GPU-only auto-fit by VRAM
"""
from controlmt.client import ControlMT
from controlmt._version import __version__
__all__ = ["ControlMT", "__version__"]