Bläddra i källkod

Update src/llama_recipes/finetuning.py

Matthias Reso 9 månader sedan
förälder
incheckning
4110202faa
1 ändrade filer med 1 tillägg och 1 borttagningar
  1. 1 1
      src/llama_recipes/finetuning.py

+ 1 - 1
src/llama_recipes/finetuning.py

@@ -101,7 +101,7 @@ def main(**kwargs):
             wandb_run = setup_wandb(train_config, fsdp_config, **kwargs)
     
     #setting quantization configs        
-    quant_config = quantizatio_config(
+    quant_config = quantization_config(
         quant_type=train_config.quantization,
         compute_dtype=torch.bfloat16,
         use_double_quant=True,