Использование 4 -битного квантования `bitsAndbytesPython

Программы на Python
Anonymous
 Использование 4 -битного квантования `bitsAndbytes

Сообщение Anonymous »

При загрузке токенизатора я получил эту ошибку: < /p>

Код: Выделить всё

ImportError: Using bitsandbytes 4-bit quantization requires the latest version of bitsandbytes:
pip install -U bitsandbytes.
< /code>
Я использую ноутбук Jupyter на MacBook M2 Pro. < /p>
Ниже приведен исходный код: < /p>
quant_config = BitsAndBytesConfig(
load_in_4bit=True,
bnb_4bit_use_double_quant=True,
bnb_4bit_compute_dtype=torch.bfloat16,
bnb_4bit_quant_type="nf4"

tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL, trust_remote_code=True)
tokenizer.pad_token = tokenizer.eos_token
tokenizer.padding_side = "right"

base_model = AutoModelForCausalLM.from_pretrained(
BASE_MODEL,
quantization_config=quant_config,
device_map="auto",
)

base_model.generation_config.pad_token_id = tokenizer.pad_token_id
Может ли кто -нибудь помочь в этом?>

Подробнее здесь: https://stackoverflow.com/questions/793 ... -bitsandby

Вернуться в «Python»