Sfoglia il codice sorgente

fix import hsdp_device_mesh

Igor Kasianenko 3 mesi fa
parent
commit
b56a04464c
1 ha cambiato i file con 0 aggiunte e 1 eliminazioni
  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,