Browse Source

fix typos

Hamid Shojanazeri 1 year ago
parent
commit
aa24e8f57e
2 changed files with 2 additions and 2 deletions
  1. 1 1
      src/llama_recipes/finetuning.py
  2. 1 1
      src/llama_recipes/utils/__init__.py

+ 1 - 1
src/llama_recipes/finetuning.py

@@ -37,7 +37,7 @@ from llama_recipes.utils.config_utils import (
 )
 from llama_recipes.utils.dataset_utils import get_preprocessed_dataset
 
-from llama_recipes.utils.fsdp_utils import hdsp_device_mesh
+from llama_recipes.utils.fsdp_utils import hsdp_device_mesh
 from llama_recipes.utils.train_utils import (
     train,
     freeze_transformer_layers,

+ 1 - 1
src/llama_recipes/utils/__init__.py

@@ -3,5 +3,5 @@
 
 from llama_recipes.utils.memory_utils import MemoryTrace
 from llama_recipes.utils.dataset_utils import *
-from llama_recipes.utils.fsdp_utils import fsdp_auto_wrap_policy, hdsp_device_mesh
+from llama_recipes.utils.fsdp_utils import fsdp_auto_wrap_policy, hsdp_device_mesh
 from llama_recipes.utils.train_utils import *