Pārlūkot izejas kodu

adding quanitzation configs

HamidShojanazeri 10 mēneši atpakaļ
vecāks
revīzija
e120d1080c
1 mainītis faili ar 1 papildinājumiem un 0 dzēšanām
  1. 1 0
      src/llama_recipes/configs/__init__.py

+ 1 - 0
src/llama_recipes/configs/__init__.py

@@ -5,3 +5,4 @@ from llama_recipes.configs.peft import lora_config, llama_adapter_config, prefix
 from llama_recipes.configs.fsdp import fsdp_config
 from llama_recipes.configs.training import train_config
 from llama_recipes.configs.wandb import wandb_config
+from llama_recipes.configs.quantization import quantizatio_config