Skip to main content

clika_runtime.modelverse.auto

Load a model by name, the way you would from a model hub.

:class:AutoModel picks the load door from the family's declared modalities; the task loaders (:class:AutoModelForCausalLM, :class:AutoModelForImageTextToText, :class:AutoModelForSpeechSeq2Seq, :class:AutoModelForTextToWaveform, :class:AutoModelForSeq2SeqLM, :class:AutoModelForObjectDetection, :class:AutoModelForZeroShotObjectDetection, :class:AutoModelForDepthEstimation) each name their door; :class:AutoTokenizer loads a checkpoint's tokenizer without its weights; :class:AutoProcessor loads its preprocessing (the tokenizer and the media processor its directory describes) without its weights; :class:AutoConfig reads a checkpoint's identity, a :class:PretrainedConfig, without its weights::

import clika_runtime.modelverse as mv

model = mv.AutoModelForCausalLM.from_pretrained(
"Qwen/Qwen2.5-0.5B-Instruct", device="cuda:0", dtype="bfloat16")
tokenizer = mv.AutoTokenizer.from_pretrained("Qwen/Qwen2.5-0.5B-Instruct")
NameKind
AutoConfigclass
AutoModelclass
AutoModelForCausalLMclass
AutoModelForDepthEstimationclass
AutoModelForImageTextToTextclass
AutoModelForObjectDetectionclass
AutoModelForSeq2SeqLMclass
AutoModelForSpeechSeq2Seqclass
AutoModelForTextToWaveformclass
AutoModelForZeroShotObjectDetectionclass
AutoProcessorclass
AutoTokenizerclass
PretrainedConfigclass
functionsmodule functions