AzzurroQuantized

Model Information

AzzurroQuantized is a compact iteration of the model Azzurro, optimized for efficiency.

It is offered in two distinct configurations: a 4-bit version and an 8-bit version, each designed to maintain the model's effectiveness while significantly reducing its size and computational requirements.

• It's trained both on publicly available datasets, like SQUAD-it, and datasets we've created in-house.
• it's designed to understand and maintain context, making it ideal for Retrieval Augmented Generation (RAG) tasks and applications requiring contextual awareness.
• It is quantized in a 4-bit version and an 8-bit version following the procedure here.

Evaluation

We evaluated the model using the same test sets as used for the Open Ita LLM Leaderboard:

Usage

You need to download the .gguf model first

If you want to use the cpu install these dependencies:

pip install llama-cpp-python huggingface_hub

If you want to use the gpu instead:

CMAKE_ARGS="-DLLAMA_CUBLAS=on" pip install huggingface_hub llama-cpp-python --force-reinstall --upgrade --no-cache-dir

And then use this code to see a response to the prompt.

from huggingface_hub import hf_hub_download
from llama_cpp import Llama

model_path = hf_hub_download(
    repo_id="MoxoffSpA/AzzurroQuantized",
    filename="Azzurro-ggml-Q4_K_M.gguf"
)

# Set gpu_layers to the number of layers to offload to GPU. Set to 0 if no GPU acceleration is available on your system.
llm = Llama(
  model_path=model_path,
  n_ctx=2048,  # The max sequence length to use - note that longer sequence lengths require much more resources
  n_threads=8,            # The number of CPU threads to use, tailor to your system and the resulting performance
  n_gpu_layers=0         # The number of layers to offload to GPU, if you have GPU acceleration available
)

# Simple inference example
question = """Quanto è alta la torre di Pisa?"""
context = """
La Torre di Pisa è un campanile del XII secolo, famoso per la sua inclinazione. Alta circa 56 metri.
"""

prompt = f"Domanda: {question}, contesto: {context}"

output = llm(
  f"[INST] {prompt} [/INST]", # Prompt
  max_tokens=128,
  stop=["\n"],   
  echo=True,
  temperature=0.1,
  top_p=0.95
)

# Chat Completion API

print(output['choices'][0]['text'])

Bias, Risks and Limitations

AzzurroQuantized and its original model Azzurro have not been aligned to human preferences for safety within the RLHF phase or deployed with in-the-loop filtering of responses like ChatGPT, so the model can produce problematic outputs (especially when prompted to do so). It is also unknown what the size and composition of the corpus were used to train the base model mistralai/Mistral-7B-v0.2, however, it is likely to have included a mix of Web data and technical sources like books and code.

Links to resources

• SQUAD-it dataset: https://huggingface.co/datasets/squad_it
• Mistral_7B_v0.2 original weights: https://models.mistralcdn.com/mistral-7b-v0-2/mistral-7B-v0.2.tar
• Mistral_7B_v0.2 model: https://huggingface.co/alpindale/Mistral-7B-v0.2-hf
• Open Ita LLM Leaderbord: https://huggingface.co/spaces/FinancialSupport/open_ita_llm_leaderboard

Base version

We have the not quantized version here: https://huggingface.co/MoxoffSpA/Azzurro

The Moxoff Team

Jacopo Abate, Marco D'Ambra, Luigi Simeone, Gianpaolo Francesco Trotta