update imports

src/llama_cookbook/configs/__init__.py

@@ -1,8 +1,8 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
 
-from llama_recipes.configs.peft import lora_config, llama_adapter_config, prefix_config
-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 quantization_config
+from llama_cookbook.configs.peft import lora_config, llama_adapter_config, prefix_config
+from llama_cookbook.configs.fsdp import fsdp_config
+from llama_cookbook.configs.training import train_config
+from llama_cookbook.configs.wandb import wandb_config
+from llama_cookbook.configs.quantization import quantization_config

src/llama_cookbook/datasets/__init__.py

@@ -3,11 +3,11 @@
 
 from functools import partial
 
-from llama_recipes.datasets.grammar_dataset.grammar_dataset import get_dataset as get_grammar_dataset
-from llama_recipes.datasets.alpaca_dataset import InstructionDataset as get_alpaca_dataset
-from llama_recipes.datasets.custom_dataset import get_custom_dataset,get_data_collator
-from llama_recipes.datasets.samsum_dataset import get_preprocessed_samsum as get_samsum_dataset
-from llama_recipes.datasets.toxicchat_dataset import get_llamaguard_toxicchat_dataset as get_llamaguard_toxicchat_dataset
+from llama_cookbook.datasets.grammar_dataset.grammar_dataset import get_dataset as get_grammar_dataset
+from llama_cookbook.datasets.alpaca_dataset import InstructionDataset as get_alpaca_dataset
+from llama_cookbook.datasets.custom_dataset import get_custom_dataset,get_data_collator
+from llama_cookbook.datasets.samsum_dataset import get_preprocessed_samsum as get_samsum_dataset
+from llama_cookbook.datasets.toxicchat_dataset import get_llamaguard_toxicchat_dataset as get_llamaguard_toxicchat_dataset
 DATASET_PREPROC = {
     "alpaca_dataset": partial(get_alpaca_dataset),
     "grammar_dataset": get_grammar_dataset,