相关项目
bitsandbytes
bitsandbytes是一个轻量级Python库，为CUDA自定义函数提供封装。该库主要提供8位优化器、矩阵乘法(LLM.int8())以及8位和4位量化功能。通过bitsandbytes.nn模块实现多位线性层，bitsandbytes.optim模块提供优化器。目前正在拓展对更多硬件后端的支持，包括Intel CPU+GPU、AMD GPU和Apple Silicon，Windows平台的支持也在开发中。
Llama-2-7b-LoRA-alpaca-cleaned
该项目客观描述了在PEFT框架下应用bitsandbytes量化配置进行模型训练的详细过程，尤其是4bit量化策略与bfloat16计算类型的使用。这些配置旨在提升模型效率，优化计算负载表现。