Ver código fonte

fix get_policies import

Igor Kasianenko 2 meses atrás
pai
commit
87da335d24
1 arquivos alterados com 1 adições e 2 exclusões
  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,