Explorar o código

Update src/llama_recipes/configs/quantization.py

Matthias Reso hai 10 meses
pai
achega
b5dc9bb13b
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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
 from transformers import BitsAndBytesConfig
 
 
 @dataclass
 @dataclass
-class quantizatio_config:
+class quantization_config:
     quant_type: str  # "int4" or "int8"
     quant_type: str  # "int4" or "int8"
     compute_dtype: torch.dtype
     compute_dtype: torch.dtype
     use_double_quant: bool = False
     use_double_quant: bool = False