|
@@ -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,
|