Преглед изворни кода

Update src/llama_recipes/configs/quantization.py

Matthias Reso пре 10 месеци
родитељ
комит
b5dc9bb13b
1 измењених фајлова са 1 додато и 1 уклоњено
  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