Kaynağa Gözat

Update src/llama_recipes/configs/quantization.py

Matthias Reso 1 yıl önce
ebeveyn
işleme
b5dc9bb13b
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  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