webscout 7.3__py3-none-any.whl → 7.5__py3-none-any.whl

webscout/Local/model.py

@@ -1,1368 +0,0 @@
-
-
-import os
-import sys
-import uuid
-import numpy as np
-
-from .utils import (
-    _SupportsWriteAndFlush,
-    UnreachableException,
-    print_version_info,
-    QuickGGUFReader,
-    print_warning,
-    print_verbose,
-    assert_type,
-    NoneType,
-    truncate,
-    softmax
-)
-
-from llama_cpp import Llama, StoppingCriteriaList
-from typing    import Generator, Optional
-from .samplers import SamplerSettings
-
-
-from webscout import exceptions
-
-
-class Model:
-    """
-    A high-level abstraction of a Llama model
-
-    The following methods are available:
-    - unload:
-        Unload the model from memory
-    - reload:
-        Re-load the model, optionally changing parameters
-    - load:
-        Load the model into memory
-    - is_loaded:
-        Return `True` if the model is fully loaded, `False` otherwise
-    - tokenize:
-        Tokenize the given text, from `str` to `list[int]`
-    - detokenize:
-        Detokenize the given text, from `list[int]` or `int` to `str`
-    - get_length:
-        Return the length of the given text as measured in tokens
-    - get_tokenization_mapping:
-        Return a mapping of token IDs to tokens for a given text
-    - print_tokenization_mapping:
-        Display the tokenization map for a given text
-    - generate:
-        Generate text from an input and return it all at once when finished
-    - stream:
-        Return a Generator that yields tokens as they are generated
-    - stream_print:
-        Stream tokens to a file as they are generated
-    - ingest:
-        Ingest the given text into the model's cache, reducing the latency of
-        future generations that start with the same text
-    - candidates:
-        Return a sorted list of candidates for the next token, along with
-        their normalized probabilities
-    - print_candidates:
-        Print a sorted list of candidates for the next token, along with
-        their normalized probabilities
-    
-    The following attributes are available:
-    - verbose `bool`:
-        Whether the model was loaded with `verbose=True`
-    - metadata `dict`:
-        A dictionary containing the GGUF metadata of the model
-    - context_length `int`:
-        The currently loaded context length of the model, in tokens
-    - n_ctx `int`:
-        Alias to context_length
-    - llama `llama_cpp.Llama`:
-        The underlying Llama instance
-    - vocab `list[str]`:
-        A list of all tokens in the model's vocabulary
-    - bos_token `int`:
-        The beginning-of-sequence token ID
-    - eos_token `int`:
-        The end-of-sequence token ID
-    - eot_token `int`:
-        The end-of-turn token ID (or `None` if not found)
-    - nl_token `int`:
-        The newline token ID (or `None` if not found)
-    - prefix_token `int`:
-        The infill prefix token ID (or `None` if not found)
-    - middle_token `int`:
-        The infill middle token ID (or `None` if not found)
-    - suffix_token `int`:
-        The infill suffix token ID (or `None` if not found)
-    - cls_token `int`:
-        The classifier token ID (or `None` if not found)
-    - sep_token `int`:
-        The separator token ID (or `None` if not found)
-    - filename `str`:
-        The name of the file the model was loaded from
-    - n_ctx_train `int`:
-        The native context length of the model
-    - rope_freq_base_train `float`:
-        The native RoPE frequency base (theta) value
-    - rope_freq_base `float`:
-        The currently loaded RoPE frequency base (theta) value
-    - flash_attn `bool`:
-        Whether the model was loaded with Flash Attention enabled
-    - n_vocab `int`:
-        The number of tokens in the model's vocabulary
-    - n_layer `int`:
-        The number of layers in the model
-    - n_gpu_layers `int`:
-        The number of layers offloaded to the GPU (-1 for all layers)
-    - type_k `int`:
-        The GGML data type used for the `K` cache. 1 == f16, q8_0 otherwise
-    - type_v `int`:
-        The GGML data type used for the `V` cache. 1 == f16, q8_0 otherwise
-    - n_gqa `int`:
-        The GQA (Grouped-Query Attention) factor of the model
-    - uuid `uuid.UUID`:
-        A randomly generated UUID, unique to this specific model instance
-    """
-
-    def __init__(
-        self,
-        model_path: str,
-        context_length: Optional[int] = 2048,
-        n_gpu_layers: int = 0,
-        offload_kqv: bool = True,
-        flash_attn: bool = False,
-        quantize_kv_cache: bool = False,
-        verbose: bool = False,
-        **kwargs
-    ):
-        """
-        Given the path to a GGUF file, construct a Model instance.
-
-        The model must be in GGUF format.
-
-        The following parameters are optional:
-        - context_length:
-            The context length at which to load the model, in tokens
-        - n_gpu_layers:
-            The number of layers to be offloaded to the GPU
-        - offload_kqv:
-            Whether the KQV cache (context) should be offloaded
-        - flash_attn:
-            Whether to use Flash Attention
-        - quantize_kv_cache:
-            Whether to use q8_0 values for KV cache
-        - verbose:
-            Whether to print additional backend information. `bool`
-
-        The following additional keyword arguments are also accepted:
-        - do_not_load:
-            If `True`, construct the model instance but do not load it into
-            memory yet. Call `Model.load()` before using the model
-        - debug:
-            If `True`, print additional backend information from llama.cpp
-        """
-
-        assert_type(verbose, bool, 'verbose', 'Model')        
-        assert_type(model_path, str, 'model_path', 'Model')
-        if not os.path.exists(model_path):
-            raise FileNotFoundError(
-                f"Model: the given model_path {model_path!r} does not exist"
-            )
-        if os.path.isdir(model_path):
-            raise IsADirectoryError(
-                f"Model: the given model_path {model_path!r} is a directory, "
-                "not a GGUF file"
-            )
-        assert_type(context_length, (int, NoneType), 'context_length', 'Model')
-        assert_type(n_gpu_layers, int, 'n_gpu_layers', 'Model')
-        assert_type(offload_kqv, bool, 'offload_kqv', 'Model')
-        assert_type(flash_attn, bool, 'flash_attn', 'Model')
-        assert_type(quantize_kv_cache, bool, 'quantize_kv_cache', 'Model')
-
-        # save __init__ parameters for __repr__
-        self._model_path = model_path
-        self._context_length = context_length
-        self._n_gpu_layers = n_gpu_layers
-        self._offload_kqv = offload_kqv
-        self._flash_attn = flash_attn
-        self._verbose = self.verbose = verbose
-        self._quantize_kv_cache = quantize_kv_cache
-
-        _kwargs_keys = kwargs.keys() # only read once
-
-        if '__uuid' not in _kwargs_keys:
-            self.uuid = uuid.uuid4()
-        else:
-            # Model.reload() passes this kwarg to preserve the UUID
-            self.uuid = kwargs.get('__uuid')
-
-        if 'do_not_load' in _kwargs_keys:
-            if kwargs.get('do_not_load') is True:
-                 # only save __init__ params to be used later in self.load()
-                return
-        
-        if verbose:
-            print_version_info(file=sys.stderr)
-
-        if sys.byteorder == 'big':
-            print_warning(
-                "host is big-endian, please ensure your GGUF file is also "
-                "big-endian"
-            )
-        elif sys.byteorder == 'little':
-            if verbose:
-                print_verbose(
-                    "host is little-endian"
-                )
-        else:
-            print_warning(
-                f"unexpected value for sys.byteorder: {sys.byteorder!r}, "
-                "expected 'little' for little-endian host or 'big' for "
-                "big-endian host"
-            )
-        
-        self._model_file_size_bytes = os.stat(model_path).st_size
-        self.metadata = QuickGGUFReader.load_metadata(model_path)
-
-        _debug = False
-        if 'debug' in _kwargs_keys:
-            _debug = bool(kwargs.get('debug'))
-
-        if verbose and not _debug:
-            __class__._print_metadata(self.metadata)
-
-        n_ctx_train = None
-        rope_freq_base_train = None
-        n_layer = None
-        n_attn_heads = None
-        n_kv_heads = None
-        n_gqa = None
-
-        for key in self.metadata.keys():
-            if key.endswith('.context_length'):
-                n_ctx_train = int(self.metadata[key])
-            elif key.endswith('.rope.freq_base'):
-                rope_freq_base_train = float(self.metadata[key])
-            elif key.endswith('.block_count'):
-                n_layer = int(self.metadata[key])
-            elif key.endswith('.attention.head_count'):
-                n_attn_heads = int(self.metadata[key])
-            elif key.endswith('.attention.head_count_kv'):
-                n_kv_heads = int(self.metadata[key])
-
-        if n_layer is None:
-            exc =  KeyError(
-                f"GGUF file metadata does not specify n_layer"
-            )
-            exc.add_note(
-                f"GGUF file is at {self._model_path!r}"
-            )
-            raise exc
-
-        if n_ctx_train is None:
-            exc =  KeyError(
-                f"GGUF file metadata does not specify a context length"
-            )
-            exc.add_note(
-                f"GGUF file is at {self._model_path!r}"
-            )
-            raise exc
-
-        if n_attn_heads is not None and n_kv_heads is not None:
-            n_gqa = int(n_attn_heads / n_kv_heads)
-        
-        if context_length <= 0:
-            context_length = None
-        
-        rope_freq_base = __class__._calculate_rope_freq_base(
-            n_ctx_train,
-            context_length if context_length is not None else n_ctx_train,
-            rope_freq_base_train
-        )
-
-        if context_length is None:
-            if n_ctx_train > 32768:
-                print_warning(
-                    f"you did not specify a context length, and the native "
-                    f"context length of this model is very large "
-                    f"({n_ctx_train}). defaulting to 32768 to avoid "
-                    f"out-of-memory errors. you should specify a higher "
-                    f"context length if you need it"
-                )
-                self.context_length = self.n_ctx = 32768
-            else:
-                self.context_length = self.n_ctx = n_ctx_train
-
-        elif context_length <= n_ctx_train:
-            self.context_length = self.n_ctx = context_length
-        
-        elif context_length > n_ctx_train:
-            print_warning(
-                f"you have specified a context length that is greater than "
-                f"the natively supported context length of this model "
-                f"({context_length} > {n_ctx_train}). the model will still "
-                f"work, but the quality of output may be subpar. consider "
-                f"decreasing the context length to {n_ctx_train} or lower "
-                f"for best results"
-            )
-            self.context_length = self.n_ctx = context_length
-        
-        else:
-            raise UnreachableException
-
-        cpu_count = int(os.cpu_count()) # only read once
-
-        if n_gpu_layers < 0 or n_gpu_layers > n_layer:
-            n_gpu_layers = n_layer
-
-        if n_gpu_layers == n_layer:
-            # fully offloaded
-            n_batch = 1024
-        else:
-            # partially offloaded
-            n_batch = 512
-        
-        # NOTE: the optimal n_threads value (for text generation) is equal
-        #       to the number of physical cores (for homogenous CPUs) or
-        #       to the number of performance cores (for heterogenous CPUs)
-        #
-        #       the optimal n_threads_batch value (for prompt eval) is equal
-        #       to the total number of logical cores, regardless of
-        #       their type
-
-        n_threads = max(cpu_count//2, 1)
-        n_threads_batch = cpu_count
-
-        if flash_attn and n_gpu_layers == 0:
-            flash_attn = False
-            print_warning(
-                "disabling flash_attn because n_gpu_layers == 0"
-            )
-        
-        if quantize_kv_cache:
-            # use q8_0 for K, V
-            if flash_attn:
-                type_k = 8
-                type_v = 8
-                if verbose:
-                   print_verbose(
-                        "using q8_0 KV cache"
-                    )
-            else:           # llama.cpp requires flash_attn for V quantization
-                type_k = 8
-                type_v = 1
-                if verbose:
-                    print_verbose(
-                        "using q8_0 K cache, f16 V cache"
-                    )
-                    print_verbose(
-                        "to quantize V cache, flash_attn must be enabled"
-                    )
-        else:
-            # use f16 for K, V (default)
-            type_k = 1
-            type_v = 1
-        
-        # guard against models with no rope_freq_base
-        if rope_freq_base is None:
-            rope_freq_base = 0
-        
-        if verbose:
-            print_verbose(
-                f"attempting to load model, offloading "
-                f"{n_gpu_layers}/{n_layer} layers..."
-            )
-        
-        # llama.cpp needs -ngl set to `-1`, not just n_layer
-        if n_gpu_layers >= n_layer:
-            _llama_ngl = -1
-        else:
-            _llama_ngl = n_gpu_layers
-
-        self.llama = Llama(
-            model_path=model_path,
-            n_ctx=self.context_length,
-            n_gpu_layers=_llama_ngl,
-            use_mmap=True,
-            use_mlock=False,
-            logits_all=False,
-            n_batch=n_batch,
-            n_threads=n_threads,
-            n_threads_batch=n_threads_batch,
-            rope_freq_base=rope_freq_base,
-            mul_mat_q=True,
-            offload_kqv=offload_kqv,
-            flash_attn=flash_attn,
-            type_k=type_k,
-            type_v=type_v,
-            verbose=_debug
-        )
-        
-        # NOTE: llama.cpp uses the nearest multiple of 32 as the actual
-        #       context length. here we update self.context_length to reflect
-        #       this
-        self.context_length = self.n_ctx = self.llama.n_ctx()
-
-        if self.n_ctx < 512:
-            print_warning(
-                f'the currently loaded context length is less than 512 tokens '
-                f'({self.n_ctx} < 512). sometimes this can cause problems in '
-                f'llama.cpp. consider increasing the context length to at '
-                f'least 512 tokens'
-            )
-
-        try:
-            self.vocab: list[str] = self.metadata['tokenizer.ggml.tokens']
-        except (KeyError, TypeError, ValueError):
-            print_warning(
-                "could not set Model.vocab, constructing manually..."
-            )
-            self.vocab = [
-                self.llama._model.detokenize([i], special=True).decode(
-                    'utf-8', errors='ignore'
-                ) for i in range(self.llama._model.n_vocab())
-            ]
-        try:
-            self.bos_token = int(self.metadata['tokenizer.ggml.bos_token_id'])
-        except (KeyError, TypeError, ValueError):
-            self.bos_token = int(self.llama._model.token_bos())
-            if self.bos_token < 0:
-                self.bos_token = None
-                print_warning(
-                    "could not set Model.bos_token, defaulting to None"
-                )
-        try:
-            self.eos_token = int(self.metadata['tokenizer.ggml.eos_token_id'])
-        except (KeyError, TypeError, ValueError):
-            self.eos_token = int(self.llama._model.token_eos())
-            if self.eos_token < 0:
-                self.eos_token = None
-                print_warning(
-                    "could not set Model.eos_token, defaulting to None"
-                )
-
-        # These special tokens are optional
-
-        self.eot_token = int(self.llama._model.token_eot())
-        if self.eot_token < 0:
-            self.eot_token = None
-
-        self.nl_token  = int(self.llama._model.token_nl())
-        if self.nl_token < 0:
-            self.nl_token = None
-
-        self.prefix_token = int(self.llama._model.token_prefix())
-        if self.prefix_token < 0:
-            self.prefix_token = None
-
-        self.middle_token = int(self.llama._model.token_middle())
-        if self.middle_token < 0:
-            self.middle_token = None
-
-        self.suffix_token = int(self.llama._model.token_suffix())
-        if self.suffix_token < 0:
-            self.suffix_token = None
-
-        self.cls_token = int(self.llama._model.token_cls())
-        if self.cls_token < 0:
-            self.cls_token = None
-
-        self.sep_token = int(self.llama._model.token_sep())
-        if self.sep_token < 0:
-            self.sep_token = None
-        
-        # Misc. attributes
-        _add_bos_token = self.llama._model.add_bos_token()
-        if _add_bos_token == 1:
-            self.add_bos_token = True
-        elif _add_bos_token == 0:
-            self.add_bos_token = False
-        else:
-            self.add_bos_token = None
-            print_warning(
-                "Model.add_bos_token is unknown, defaulting to None"
-            )
-        
-        _add_eos_token = self.llama._model.add_eos_token()
-        if _add_eos_token == 1:
-            self.add_eos_token = True
-        elif _add_eos_token == 0:
-            self.add_eos_token = False
-        else:
-            self.add_eos_token = None
-            print_warning(
-                "Model.add_eos_token is unknown, defaulting to None"
-            )
-
-        self.filename: str = os.path.basename(model_path)
-        self.n_ctx_train: int = n_ctx_train
-        self.rope_freq_base_train: float = rope_freq_base_train
-        self.rope_freq_base: float = rope_freq_base
-        self.n_batch: int = n_batch
-        self.n_threads: int = n_threads
-        self.n_threads_batch: int = n_threads_batch
-        self.flash_attn: bool = flash_attn
-        self.n_embd = self.llama._model.n_embd()
-        self.n_params = self.llama._model.n_params()
-        self.bpw = (8*self._model_file_size_bytes)/self.n_params
-        self.n_vocab: int = len(self.vocab)
-        self.n_layer: int = n_layer
-        self.n_gpu_layers: int = n_gpu_layers
-        self.offload_kqv = offload_kqv
-        self.is_native: bool = self.context_length <= self.n_ctx_train
-        self.type_k: int = type_k
-        self.type_v: int = type_v
-        self.n_gqa: int = n_gqa
-
-        if verbose:
-            print_verbose(
-                f"{'new' if '__uuid' not in _kwargs_keys else 'reloaded'} "
-                f"Model instance with the following attributes:"
-            )
-            print_verbose(f"   uuid                 == {self.uuid}")
-            print_verbose(f"   filename             == {self.filename}")
-            print_verbose(f"   n_params             == {self.n_params}")
-            print_verbose(
-                f"   bpw                  == {self.bpw} "
-                f"({__class__._get_bpw_quality_hint(self.bpw)})"
-            )
-            print_verbose(f"   n_gpu_layers         == {self.n_gpu_layers}")
-            print_verbose(f"   n_layer              == {self.n_layer}")
-            print_verbose(f"   offload_kqv          == {self.offload_kqv}")
-            print_verbose(f"   flash_attn           == {self.flash_attn}")
-            print_verbose(f"   n_gqa                == {self.n_gqa}")
-            print_verbose(
-                f"   type_k               == {self.type_k} "
-                f"({'f16' if self.type_k == 1 else 'q8_0'})"
-            )
-            print_verbose(
-                f"   type_v               == {self.type_v} "
-                f"({'f16' if self.type_v == 1 else 'q8_0'})"
-            )
-            print_verbose(f"   n_batch              == {self.n_batch}")
-            print_verbose(
-                f"   n_threads            == {self.n_threads}/{cpu_count}"
-            )
-            print_verbose(
-                f"   n_threads_batch      == {self.n_threads_batch}/{cpu_count}"
-            )
-            print_verbose(f"   n_ctx_train          == {self.n_ctx_train}")
-            print_verbose(f"   n_ctx                == {self.n_ctx}")
-            print_verbose(f"   rope_freq_base_train == {self.rope_freq_base_train}")
-            print_verbose(f"   rope_freq_base       == {self.rope_freq_base}")
-            print_verbose(f"   n_embd               == {self.n_embd}")
-            print_verbose(f"   n_vocab              == {self.n_vocab}")
-            print_verbose(f"   bos_token            == {self.bos_token}")
-            print_verbose(f"   eos_token            == {self.eos_token}")
-            if self.eot_token is not None:
-                print_verbose(f"   eot_token            == {self.eot_token}")
-            if self.nl_token is not None:
-                print_verbose(f"   nl_token             == {self.nl_token}")
-            if self.prefix_token is not None:
-                print_verbose(f"   prefix_token         == {self.prefix_token}")
-            if self.middle_token is not None:
-                print_verbose(f"   middle_token         == {self.middle_token}")
-            if self.suffix_token is not None:
-                print_verbose(f"   suffix_token         == {self.suffix_token}")
-            if self.cls_token is not None:
-                print_verbose(f"   cls_token            == {self.cls_token}")
-            if self.sep_token is not None:
-                print_verbose(f"   sep_token            == {self.sep_token}")
-            print_verbose(f"   add_bos_token        == {self.add_bos_token}")
-            print_verbose(f"   add_eos_token        == {self.add_eos_token}")
-
-
-    @staticmethod
-    def _calculate_rope_freq_base(
-            n_ctx_train: int,
-            n_ctx_load: int,
-            rope_freq_base_train: Optional[float]
-        ) -> float:
-        """
-        Returns the rope_freq_base (theta) value at which model should be loaded
-        """
-        assert_type(n_ctx_train, int, 'n_ctx_train', '_calculate_rope_freq_base')
-        assert_type(n_ctx_load, int, 'n_ctx_load', '_calculate_rope_freq_base')
-        assert_type(rope_freq_base_train, (float, NoneType),
-                    'rope_freq_base_train', '_calculate_rope_freq_base')
-
-        if n_ctx_load <= n_ctx_train:
-            if rope_freq_base_train is None:
-                return 0.0
-            else:
-                return rope_freq_base_train
-        
-        if rope_freq_base_train is None or rope_freq_base_train == 0.0:
-            raise ValueError(
-                'unable to load model with greater than native '
-                f'context length ({n_ctx_load} > {n_ctx_train}) '
-                'because model does not specify rope_freq_base. '
-                f'try again with context_length <= {n_ctx_train}'
-            )
-        
-        return ((n_ctx_load/n_ctx_train)**(2**(1/4)))*rope_freq_base_train
-
-        # traditional formula:
-        #   return (n_ctx_load/n_ctx_train)*rope_freq_base_train
-        # experimental formula A:
-        #   return ((n_ctx_load/n_ctx_train)**2)*rope_freq_base_train
-        # experimental formula B:
-        #   return ((n_ctx_load/n_ctx_train)**(2**(1/4)))*rope_freq_base_train
-    
-
-    @staticmethod
-    def _get_bpw_quality_hint(bpw: float) -> str:
-        if 0.0 < bpw < 2.0:
-            return 'terrible'
-        elif 2.0 <= bpw < 4.0:
-            return 'bad'
-        elif 4.0 <= bpw < 5.0:
-            return 'good'
-        elif 5.0 <= bpw < 16.0:
-            return 'great'
-        elif bpw >= 16.0:
-            return 'native'
-        else:
-            raise UnreachableException
-    
-
-    @staticmethod
-    def _print_metadata(
-        metadata: dict,
-        file: _SupportsWriteAndFlush = sys.stderr
-    ) -> None:
-        max_len_key = max(len(k) for k in metadata.keys())
-        print(f'webscout.Local: read model metadata from GGUF file header:', file=file)
-        for k, v in metadata.items():
-            print(
-                f'webscout.Local:    {k:<{max_len_key}} : {truncate(repr(v))}',
-                file=file
-            )
-
-    
-    def __repr__(self) -> str:
-        return (
-            f"Model({self._model_path!r}, "
-            f"context_length={self._context_length}, "
-            f"n_gpu_layers={self._n_gpu_layers}, "
-            f"offload_kqv={self._offload_kqv}, "
-            f"flash_attn={self._flash_attn}, "
-            f"quantize_kv_cache={self._quantize_kv_cache}, "
-            f"verbose={self._verbose})"
-        )
-    
-    
-    def __sizeof__(self) -> int:
-        """Returns the size of the model file on disk, NOT the memory usage"""
-        return self._model_file_size_bytes
-    
-
-    def __del__(self):
-       if self.is_loaded():
-        self.unload()
-    
-
-    def __enter__(self):
-        if not self.is_loaded():
-            self.load()
-        return self
-
-
-    def __exit__(self, *_):
-        if self.is_loaded():
-            self.unload()
-    
-
-    def __call__(
-        self,
-        prompt: str | list[int],
-        stops: Optional[list[str | int]] = None,
-        sampler: Optional[SamplerSettings] = None
-    ) -> str:
-        """
-        `Model(...)` is a shorthand for `Model.generate(...)`
-        """
-        return self.generate(prompt=prompt, stops=stops, sampler=sampler)
-    
-
-    def __eq__(self, value: object, /) -> bool:
-        if not isinstance(value, __class__):
-            return NotImplemented
-        if not (hasattr(self, 'uuid') and hasattr(value, 'uuid')):
-            raise AttributeError(
-                "At least one of the models being compared is missing the "
-                "`.uuid` attribute"
-            )
-        return self.uuid == value.uuid
-    
-
-    def __hash__(self, /) -> int:
-        return hash(self.uuid)
-    
-
-    def unload(self):
-        """
-        Unload the model from memory
-
-        Does nothing if the model is not loaded
-        """
-        if not self.is_loaded():
-            if self.verbose:
-                print_verbose('model already unloaded')
-            return
-        
-        if self.verbose:
-            print_verbose('unloading model...')
-
-        self.llama.close()
-
-        while hasattr(self, 'llama'):
-            delattr(self, 'llama')
-
-        if self.verbose:
-            print_verbose('model unloaded')
-    
-
-    def reload(
-        self,
-        context_length: Optional[int] = None,
-        n_gpu_layers: Optional[int] = None,
-        offload_kqv: Optional[bool] = None,
-        flash_attn: Optional[bool] = None,
-        quantize_kv_cache: Optional[bool] = None,
-        verbose: Optional[bool] = None
-    ):
-        """
-        Re-load the model into memory using the specified parameters
-
-        Any parameters unspecified will be unchanged
-        """
-        __uuid = self.uuid
-        self.unload()
-        self.__init__(
-            model_path = self._model_path,
-            context_length = (
-                self._context_length if context_length is None
-                else context_length
-            ),
-            n_gpu_layers = (
-                self._n_gpu_layers if n_gpu_layers is None
-                else n_gpu_layers
-            ),
-            offload_kqv = (
-                self._offload_kqv if offload_kqv is None
-                else offload_kqv
-            ),
-            flash_attn = (
-                self._flash_attn if flash_attn is None
-                else flash_attn
-            ),
-            quantize_kv_cache = (
-                self._quantize_kv_cache if quantize_kv_cache is None
-                else quantize_kv_cache
-            ),
-            verbose = (
-                self._verbose if verbose is None
-                else verbose
-            ),
-            __uuid = __uuid  # do not change UUID on reload
-        )
-        assert_model_is_loaded(self)
-    
-
-    def load(self) -> None:
-        """
-        Load the model into memory
-
-        Does nothing if already loaded
-        """
-        if self.is_loaded():
-            if self.verbose:
-                print_verbose('model already loaded')
-        else:
-            self.reload()
-    
-
-    def is_loaded(self) -> bool:
-        """
-        Return `True` if the model is fully loaded, `False` otherwise
-        """
-        try:
-            assert_model_is_loaded(self)
-        except exceptions.ModelUnloadedException:
-            return False
-        else:
-            return True
-    
-
-    def tokenize(self, text: str) -> list[int]:
-        """
-        Tokenize the given text (from `str` to `list[int]`)
-        """
-        assert_type(text, str, 'text', 'tokenize')
-        assert_model_is_loaded(self)
-        tokens = self.llama._model.tokenize(
-            text.encode('utf-8'),
-            add_bos=(
-                self.add_bos_token if self.add_bos_token is not None
-                else True
-            ),
-            special=True
-        )
-        # remove duplicate BOS tokens at the start of the text
-        while len(tokens) >= 2 and tokens[0] == self.bos_token and tokens[1] == self.bos_token:
-            tokens.pop(0)
-            if self.verbose:
-                print_verbose("tokenize: removed duplicate BOS token")
-        # remove duplicate EOS tokens at the end of the text
-        while len(tokens) >= 2 and tokens[-1] == self.eos_token and tokens[-2] == self.eos_token:
-            tokens.pop(-1)
-            if self.verbose:
-                print_verbose("tokenize: removed duplicate EOS token")
-        return tokens
-
-
-    def detokenize(self, tokens: list[int] | int) -> str:
-        """
-        Detokenize the given text (from `int` or `list[int]` to `str`)
-        """
-        assert_type(tokens, (list, int), 'tokens', 'detokenize')
-        if isinstance(tokens, int):
-            tokens = [tokens]  # handle single tokens
-        for tok_id in tokens:
-            if not 0 <= tok_id < self.n_vocab:
-                raise ValueError(
-                    f"detokenize: token id {tok_id} is out of range. "
-                    f"acceptable values for this model are between 0 and "
-                    f"{self.n_vocab-1} inclusive"
-                )
-        # remove duplicate BOS tokens at the start of the text
-        while len(tokens) >= 2 and tokens[0] == self.bos_token and tokens[1] == self.bos_token:
-            tokens.pop(0)
-            if self.verbose:
-                print_verbose("detokenize: removed duplicate BOS token")
-        # remove duplicate EOS tokens at the end of the text
-        while len(tokens) >= 2 and tokens[-1] == self.eos_token and tokens[-2] == self.eos_token:
-            tokens.pop(-1)
-            if self.verbose:
-                print_verbose("detokenize: removed duplicate EOS token")
-        assert_model_is_loaded(self)
-        return self.llama._model.detokenize(
-            tokens,
-            special=True
-        ).decode('utf-8', errors='ignore')
-
-
-    def get_length(self, text: str) -> int:
-        """
-        Return the length of the given text in as measured in tokens
-        """
-        return len(self.tokenize(text))
-    
-
-    def get_tokenization_mapping(
-            self,
-            text: str
-        ) -> list[tuple[int, str]]:
-        """
-        Tokenize the given text and return a list of tuples where the first
-        item in the tuple is the token ID and the second item is the
-        corresponding text
-        """
-        token_id_list: list[int] = self.tokenize(text)
-
-        return list(
-            zip(
-                token_id_list,
-                [self.detokenize(tok_id) for tok_id in token_id_list]
-            )
-        )
-    
-
-    def print_tokenization_mapping(self, text: str) -> None:
-        """
-        Tokenize the given text and display a mapping of each
-        token ID and its corresponding decoded text
-
-        This is meant to be equivalent to `llama.cpp/llama-tokenize`
-        """
-        token_mapping_list = self.get_tokenization_mapping(text)
-
-        for token_id, token_text in token_mapping_list:
-            print(f"{token_id:>7} -> '{token_text}'")
-        print(f"Total number of tokens: {len(token_mapping_list)}")
-        
-    
-    def generate(
-        self,
-        prompt: str | list[int],
-        stops: Optional[list[str | int]] = None,
-        sampler: Optional[SamplerSettings] = None
-    ) -> str:
-        """
-        Given a prompt, return a generated string.
-
-        prompt: The text from which to generate
-
-        The following parameters are optional:
-        - stops: A list of strings and/or token IDs at which to end the generation early
-        - sampler: The SamplerSettings object used to control text generation
-        """
-        
-        stops = [] if stops is None else stops
-        assert_type(stops, list, 'stops', 'generate')
-        for item in stops:
-            assert_type(
-                item,
-                (str, int),
-                "some item in parameter 'stops'",
-                'generate'
-            )
-        
-        sampler = SamplerSettings() if sampler is None else sampler
-
-        if sampler.temp < 0.0:
-            print_warning(
-                f'generate: using negative temperature value {sampler.temp}'
-            )
-        
-        assert_type(prompt, (str, list), 'prompt', 'generate')
-        if isinstance(prompt, list):
-            prompt_tokens = prompt
-        else:
-            if self.verbose:
-                print_verbose(
-                    "generate: tokenizing prompt"
-                )
-            prompt_tokens = self.tokenize(prompt)
-        
-        input_length = len(prompt_tokens)
-
-        if input_length > self.context_length:
-            print(f'webscout.Local: raw input: {prompt_tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"generate: length of input exceeds model's context length "
-                f"({input_length} > {self.context_length})"
-            )
-        elif input_length == self.context_length:
-            print(f'webscout.Local: raw input: {prompt_tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"generate: length of input is equal to model's context "
-                f"length ({input_length} == {self.context_length}). this "
-                f"leaves no room for any new tokens to be generated"
-            )
-        elif self.verbose:
-            print_verbose(
-                f"generate: received prompt with {input_length} tokens"
-            )
-
-        stop_strs: list[str] = [stop for stop in stops if isinstance(stop, str)]
-        stop_token_ids: list[int] = [tok_id for tok_id in stops if isinstance(tok_id, int)]
-        stopping_criteria = None
-        if stop_token_ids != []:
-            def stop_on_token_ids(tokens, *args, **kwargs):
-                return tokens[-1] in stop_token_ids
-            stopping_criteria = StoppingCriteriaList([stop_on_token_ids])
-
-        if self.verbose:
-            print_verbose(f'generate: using the following sampler settings:')
-            print_verbose(f'max_len_tokens    == {sampler.max_len_tokens}')
-            print_verbose(f'top_k             == {sampler.top_k}')
-            print_verbose(f'top_p             == {sampler.top_p}')
-            print_verbose(f'min_p             == {sampler.min_p}')
-            print_verbose(f'temp              == {sampler.temp}')
-            print_verbose(f'frequency_penalty == {sampler.frequency_penalty}')
-            print_verbose(f'presence_penalty  == {sampler.presence_penalty}')
-            print_verbose(f'repeat_penalty    == {sampler.repeat_penalty}')
-
-        assert_model_is_loaded(self)
-        return self.llama.create_completion(
-            prompt=prompt_tokens,
-            max_tokens=sampler.max_len_tokens,
-            temperature=sampler.temp,
-            top_p=sampler.top_p,
-            min_p=sampler.min_p,
-            frequency_penalty=sampler.frequency_penalty,
-            presence_penalty=sampler.presence_penalty,
-            repeat_penalty=sampler.repeat_penalty,
-            top_k=sampler.top_k,
-            stop=stop_strs,
-            stopping_criteria=stopping_criteria
-        )['choices'][0]['text']
-    
-
-    def stream(
-        self,
-        prompt: str | list[int],
-        stops: Optional[list[str | int]] = None,
-        sampler: Optional[SamplerSettings] = None
-    ) -> Generator:
-        """
-        Given a prompt, return a Generator that yields dicts containing tokens.
-
-        To get the token string itself, subscript the dict with:
-
-        `['choices'][0]['text']`
-
-        prompt: The text from which to generate
-
-        The following parameters are optional:
-        - stops: A list of strings and/or token IDs at which to end the generation early
-        - sampler: The SamplerSettings object used to control text generation
-        """
-
-        stops = [] if stops is None else stops
-        assert_type(stops, list, 'stops', 'stream')
-        for item in stops:
-            assert_type(
-                item,
-                (str, int),
-                "some item in parameter 'stops'",
-                'stream'
-            )
-        
-        sampler = SamplerSettings() if sampler is None else sampler
-
-        if sampler.temp < 0.0:
-            print_warning(
-                f'stream: using negative temperature value {sampler.temp}'
-            )
-        
-        assert_type(prompt, (str, list), 'prompt', 'stream')
-        if isinstance(prompt, list):
-            prompt_tokens = prompt
-        else:
-            if self.verbose:
-                print_verbose(
-                    "stream: tokenizing prompt"
-                )
-            prompt_tokens = self.tokenize(prompt)
-        
-        input_length = len(prompt_tokens)
-
-        if input_length > self.context_length:
-            print(f'webscout.Local: raw input: {prompt_tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"stream: length of input exceeds model's context length "
-                f"({input_length} > {self.context_length})"
-            )
-        elif input_length == self.context_length:
-            print(f'webscout.Local: raw input: {prompt_tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"stream: length of input is equal to model's context "
-                f"length ({input_length} == {self.context_length}). this "
-                f"leaves no room for any new tokens to be generated"
-            )
-        elif self.verbose:
-            print_verbose(
-                f"stream: received prompt with {input_length} tokens"
-            )
-
-        stop_strs: list[str] = [stop for stop in stops if isinstance(stop, str)]
-        stop_token_ids: list[int] = [tok_id for tok_id in stops if isinstance(tok_id, int)]
-        stopping_criteria = None
-        if stop_token_ids != []:
-            def stop_on_token_ids(tokens, *args, **kwargs):
-                return tokens[-1] in stop_token_ids
-            stopping_criteria = StoppingCriteriaList([stop_on_token_ids])
-
-        if self.verbose:
-            print_verbose(f'stream: using the following sampler settings:')
-            print_verbose(f'max_len_tokens    == {sampler.max_len_tokens}')
-            print_verbose(f'top_k             == {sampler.top_k}')
-            print_verbose(f'top_p             == {sampler.top_p}')
-            print_verbose(f'min_p             == {sampler.min_p}')
-            print_verbose(f'temp              == {sampler.temp}')
-            print_verbose(f'frequency_penalty == {sampler.frequency_penalty}')
-            print_verbose(f'presence_penalty  == {sampler.presence_penalty}')
-            print_verbose(f'repeat_penalty    == {sampler.repeat_penalty}')
-        
-        assert_model_is_loaded(self)
-        return self.llama.create_completion(
-            prompt=prompt_tokens,
-            max_tokens=sampler.max_len_tokens,
-            temperature=sampler.temp,
-            top_p=sampler.top_p,
-            min_p=sampler.min_p,
-            frequency_penalty=sampler.frequency_penalty,
-            presence_penalty=sampler.presence_penalty,
-            repeat_penalty=sampler.repeat_penalty,
-            top_k=sampler.top_k,
-            stream=True,
-            stop=stop_strs,
-            stopping_criteria=stopping_criteria
-        )
-
-
-    def stream_print(
-        self,
-        prompt: str | list[int],
-        stops: Optional[list[str | int]] = None,
-        sampler: Optional[SamplerSettings] = None,
-        end: str = '\n',
-        file: _SupportsWriteAndFlush = None,
-        flush: bool = True
-    ) -> str:
-        """
-        Given a prompt, stream text to a file as it is generated, and return
-        the generated string. The returned string does not include the `end`
-        parameter.
-
-        prompt: The text from which to generate
-
-        The following parameters are optional:
-        - stops: A list of strings and/or token IDs at which to end the generation early
-        - sampler: The SamplerSettings object used to control text generation
-        - end: A string to print after the generated text
-        - file: The file where text should be printed
-        - flush: Whether to flush the stream after each token
-        """
-        
-        token_generator = self.stream(
-            prompt=prompt,
-            stops=stops,
-            sampler=sampler
-        )
-
-        file = sys.stdout if file is None else file
-
-        response = ''
-        for i in token_generator:
-            tok = i['choices'][0]['text']
-            print(tok, end='', file=file, flush=flush)
-            response += tok
-
-        # print `end`, and always flush stream after generation is done
-        print(end, end='', file=file, flush=True)
-
-        return response
-
-
-    def ingest(self, text: str | list[int]) -> None:
-        """
-        Ingest the given text into the model's cache
-        """
-
-        assert_type(text, (str, list), 'prompt', 'stream')
-        if isinstance(text, list):
-            tokens = text
-        else:
-            if self.verbose:
-                print_verbose(
-                    "ingest: tokenizing text"
-                )
-            tokens = self.tokenize(text)
-        
-        input_length = len(tokens)
-
-        if input_length > self.context_length:
-            print(f'webscout.Local: raw input: {tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"ingest: length of input exceeds model's context length "
-                f"({input_length} > {self.context_length})"
-            )
-        elif input_length == self.context_length:
-            print(f'webscout.Local: raw input: {tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"ingest: length of input is equal to model's context "
-                f"length ({input_length} == {self.context_length}). this "
-                f"leaves no room for any new tokens to be generated"
-            )
-        elif self.verbose:
-            print_verbose(
-                f"ingest: ingesting {input_length} tokens"
-            )
-        
-        assert_model_is_loaded(self)
-        self.llama.create_completion(
-            prompt=tokens,
-            max_tokens=2,
-            temperature=0.0
-        )
-
-
-    def candidates(
-        self,
-        prompt: str,
-        k: int = 40,
-        temp: Optional[float] = None,
-        raw_token_ids: bool = False
-    ) -> list[tuple[str, np.floating]]:
-        """
-        Given prompt `str` and k `int`, return a sorted list of the
-        top k candidates for most likely next token, along with their
-        normalized probabilities (logprobs).
-
-        The following parameters are optional:
-        - temp: The temperature to apply to the distribution
-        - raw_token_ids: If `True`, return raw token IDs instead of text tokens
-
-        If parameter `k` is <= 0, the probabilities for all tokens in the
-        vocabulary will be returned. Vocabulary sizes are often in the
-        hundred-thousands.
-        """
-
-        assert_type(prompt, str, 'prompt', 'candidates')
-        assert_type(k, int, 'k', 'candidates')
-        assert_type(temp, (float, NoneType), 'temp', 'candidates')
-        assert_model_is_loaded(self)
-        if k <= 0:
-            k = self.n_vocab
-            if self.verbose:
-                print_verbose(
-                    f"candidates: k <= 0, using n_vocab ({self.n_vocab})"
-                )
-        if not 1 <= k <= self.n_vocab:
-            raise ValueError(
-                f"candidates: k should be between 1 and {self.n_vocab} "
-                f"inclusive"
-            )
-
-        prompt_tokens = self.tokenize(prompt)
-        input_length = len(prompt_tokens)
-
-        if input_length > self.context_length:
-            print(f'webscout.Local: raw input: {prompt_tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"candidates: length of input exceeds model's context length "
-                f"({input_length} > {self.context_length})"
-            )
-        elif input_length == self.context_length:
-            print(f'webscout.Local: raw input: {prompt_tokens}')
-            raise exceptions.ExceededContextLengthException(
-                f"candidates: length of input is equal to model's context "
-                f"length ({input_length} == {self.context_length}). this "
-                f"leaves no room for any new tokens to be generated"
-            )
-
-        # it is necessary to reset the model before calling llama.eval()
-        elif self.verbose:
-            print_verbose(
-                "candidates: reset model state..."
-            )
-        self.llama.reset()
-
-        if self.verbose:
-            print_verbose(
-                "candidates: eval..."
-            )
-        self.llama.eval(prompt_tokens) # single forward pass
-        
-        scores = self.llama.scores[len(prompt_tokens) - 1]
-
-        # Get the top k indices based on raw scores
-        top_k_indices = np.argpartition(scores, -k)[-k:]
-
-        # Get the scores of the top k tokens
-        top_k_scores = scores[top_k_indices]
-
-        # Apply softmax to the top k scores
-        if self.verbose:
-            print_verbose(
-                f'candidates: compute softmax over {len(top_k_scores)} '
-                f'values...'
-            )
-        normalized_scores = softmax(z=top_k_scores, T=temp)
-
-        # consider only the top k tokens
-        logprobs = [
-            (
-                self.llama._model.detokenize(
-                    [tok_id], special=True
-                ).decode('utf-8', errors='ignore'),
-                normalized_scores[i]
-            ) for i, tok_id in enumerate(top_k_indices)
-        ] if not raw_token_ids else [
-            (
-                tok_id,
-                normalized_scores[i]
-            ) for i, tok_id in enumerate(top_k_indices)
-        ]
-
-        # sort by probability
-        logprobs.sort(key=lambda x: x[1], reverse=True)
-
-        return logprobs
-
-
-    def print_candidates(
-        self,
-        prompt: str,
-        k: int = 40,
-        temp: Optional[float] = None,
-        raw_token_ids: bool = False,
-        file: _SupportsWriteAndFlush = None,
-    ) -> None:
-        """
-        Given prompt `str` and k `int`, print a sorted list of the
-        top k candidates for most likely next token, along with their
-        normalized probabilities (logprobs).
-
-        The following parameters are optional:
-        - temp: The temperature to apply to the distribution
-        - raw_token_ids: If `True`, print raw token IDs instead of text tokens
-
-        If parameter `k` is <= 0, the probabilities for all tokens in the
-        vocabulary will be printed. Vocabulary sizes are often in the
-        hundred-thousands.
-        """
-        for _tuple in self.candidates(
-            prompt=prompt, k=k, temp=temp, raw_token_ids=raw_token_ids
-        ):
-            percent_as_string = f"{_tuple[1] * 100 :>7.3f}"
-            # do not print tokens with ~0.000% probability
-            if percent_as_string != "  0.000":
-                print(
-                    f"token {_tuple[0]!r:<32} has probability "
-                    f"{percent_as_string} %",
-                    file=sys.stdout if file is None else file,
-                )
-
-
-def assert_model_is_loaded(model) -> None:
-    """
-    Ensure the model is fully constructed, such that
-    `model.llama._model.model is not None` is guaranteed to be `True`
-
-    Raise ModelUnloadedException otherwise
-    """
-    try:
-        if model.llama._model.model is not None:
-            return
-    except AttributeError:
-        pass
-
-    if model is None:
-        exc = exceptions.ModelUnloadedException(
-            "model is None"
-        )
-    elif not hasattr(model, 'llama'):
-        exc = exceptions.ModelUnloadedException(
-            "webscout.Local.Model instance has no attribute 'llama'"
-        )
-    elif not hasattr(model.llama, '_model'):
-        exc = exceptions.ModelUnloadedException(
-            "llama_cpp.Llama instance has no attribute '_model'"
-        )
-    elif not hasattr(model.llama._model, 'model'):
-        exc = exceptions.ModelUnloadedException(
-            "llama_cpp._internals._LlamaModel instance has no attribute "
-            "'model'"
-        )
-    elif model.llama._model.model is None:
-        exc = exceptions.ModelUnloadedException(
-            "llama_cpp._internals._LlamaModel.model is None"
-        )
-    else:
-        raise UnreachableException
-    
-    if not isinstance(model, Model):
-        exc.add_note(
-            'WARNING: `assert_model_is_loaded` was called on an object '
-            'that is NOT an instance of `webscout.Local.Model` '
-            f'(object had type {type(model)!r})'
-        )
-    else:
-        exc.add_note(
-            'Are you trying to use a model that has been unloaded?'
-        )
-    
-    raise exc