oobabooga/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2025-12-06 07:12:10 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add the code for 4-bit quantization

Browse source
This commit is contained in:
oobabooga 2025-11-27 18:29:32 -08:00
parent 742db85de0
commit cecb172d2c
1 changed files with 11 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

11
modules/image_models.py
View file

@ -36,6 +36,17 @@ def load_image_model(model_name, dtype='bfloat16', attn_backend='sdpa', cpu_offl
quant_kwargs={"load_in_8bit": True}, quant_kwargs={"load_in_8bit": True},
) )
# Define quantization config for 4-bit
# pipeline_quant_config = PipelineQuantizationConfig(
# quant_backend="bitsandbytes_4bit",
# quant_kwargs={
# "load_in_4bit": True,
# "bnb_4bit_quant_type": "nf4", # Or "fp4" for floating point
# "bnb_4bit_compute_dtype": torch.bfloat16, # For faster computation
# "bnb_4bit_use_double_quant": True, # Nested quantization for extra savings
# },
# )
pipe = ZImagePipeline.from_pretrained( pipe = ZImagePipeline.from_pretrained(
str(model_path), str(model_path),
quantization_config=pipeline_quant_config, quantization_config=pipeline_quant_config,