Skip to content

Chapter 12: bitsandbytes error #51

@avafor

Description

@avafor

Running this part of the code in chapter 12, I get the following error. Although I have the latest version of bitsandbytes.

`from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig

model_name = "TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T"

bnb_config = BitsAndBytesConfig(
load_in_4bit=True, # Use 4-bit precision model loading
bnb_4bit_quant_type="nf4", # Quantization type
bnb_4bit_compute_dtype="float16", # Compute dtype
bnb_4bit_use_double_quant=True, # Apply nested quantization
)

model = AutoModelForCausalLM.from_pretrained(
model_name,
device_map="auto",

# Leave this out for regular SFT
quantization_config=bnb_config,

)
model.config.use_cache = False
model.config.pretraining_tp = 1

tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
tokenizer.pad_token = ""
tokenizer.padding_side = "left"`

71 raise ImportError( 72 f"Usingbitsandbytes4-bit quantization requires Accelerate:pip install 'accelerate>={ACCELERATE_MIN_VERSION}'" 73 ) 74 if not is_bitsandbytes_available(): ---> 75 raise ImportError( 76 "Using bitsandbytes4-bit quantization requires the latest version of bitsandbytes:pip install -U bitsandbytes`"
77 )
79 from ..integrations import validate_bnb_backend_availability
80 from ..utils import is_bitsandbytes_multi_backend_available

ImportError: Using bitsandbytes 4-bit quantization requires the latest version of bitsandbytes: `pip install -U bitsandbytes``

Do you have any suggestions on how to debug the issue?
Thank you very much in advance.

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions