Views
No views yet
1from transformers import BitsAndBytesConfig
2
3quantization_config = BitsAndBytesConfig(
4 load_in_4bit=True,
5 bnb_4bit_quant_type="nf4",
6 bnb_4bit_compute_dtype="float16"
7)1from transformers import AutoModel, AutoTokenizer, BitsAndBytesConfig
2
3quantization_config = BitsAndBytesConfig(
4 load_in_4bit=True,
5 bnb_4bit_quant_type="nf4",
6 bnb_4bit_compute_dtype="float16"
7)
8
9model = AutoModel.from_pretrained(
10 "Rainnighttram/Dream-7B-bnb-4bit",
11 quantization_config=quantization_config,
12 trust_remote_code=True
13)
14tokenizer = AutoTokenizer.from_pretrained(
15 "Rainnighttram/Dream-7B-bnb-4bit",
16 trust_remote_code=True
17)