Browse Source

Update src/llama_recipes/configs/quantization.py

Matthias Reso 10 months ago
parent
commit
b5dc9bb13b
1 changed files with 1 additions and 1 deletions
  1. 1 1
      src/llama_recipes/configs/quantization.py

+ 1 - 1
src/llama_recipes/configs/quantization.py

@@ -7,7 +7,7 @@ import torch
 from transformers import BitsAndBytesConfig
 
 @dataclass
-class quantizatio_config:
+class quantization_config:
     quant_type: str  # "int4" or "int8"
     compute_dtype: torch.dtype
     use_double_quant: bool = False