Browse Source

fix import hsdp_device_mesh

Igor Kasianenko 3 tháng trước cách đây
mục cha
commit
b56a04464c
1 tập tin đã thay đổi với 0 bổ sung1 xóa
  1. 0 1
      src/llama_cookbook/finetuning.py

+ 0 - 1
src/llama_cookbook/finetuning.py

@@ -39,7 +39,6 @@ from llama_cookbook.utils.train_utils import (
     clear_gpu_cache,
     freeze_transformer_layers,
     freeze_LLM_only,
-    hsdp_device_mesh,
     print_model_size,
     print_frozen_model_status,
     setup,