Преглед на файлове

fix get_policies import

Igor Kasianenko преди 2 месеца
родител
ревизия
87da335d24
променени са 1 файла, в които са добавени 1 реда и са изтрити 2 реда
  1. 1 2
      src/llama_cookbook/finetuning.py

+ 1 - 2
src/llama_cookbook/finetuning.py

@@ -34,12 +34,11 @@ from llama_cookbook.utils.dataset_utils import (
     get_preprocessed_dataset,
 )
 
-from llama_cookbook.utils.fsdp_utils import hsdp_device_mesh
+from llama_cookbook.utils.fsdp_utils import hsdp_device_mesh, get_policies
 from llama_cookbook.utils.train_utils import (
     clear_gpu_cache,
     freeze_transformer_layers,
     freeze_LLM_only,
-    get_policies,
     hsdp_device_mesh,
     print_model_size,
     print_frozen_model_status,