瀏覽代碼

fix import

Igor Kasianenko 2 月之前
父節點
當前提交
cbd2b246ba
共有 1 個文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/llama_cookbook/utils/fsdp_utils.py

+ 1 - 1
src/llama_cookbook/utils/fsdp_utils.py

@@ -5,7 +5,7 @@ import torch
 import torch.cuda.nccl as nccl
 import torch.distributed as dist
 
-from llama_recipes.policies import fpSixteen,bfSixteen, get_llama_wrapper
+from llama_cookbook.policies import fpSixteen,bfSixteen, get_llama_wrapper
 from torch.distributed._tensor.device_mesh import init_device_mesh
 
 def fsdp_auto_wrap_policy(model, transformer_layer_names):