浏览代码

update checkpointing

Sanyam Bhutani 2 月之前
父节点
当前提交
df5f969559
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 1 1
      src/llama_cookbook/model_checkpointing/__init__.py

+ 1 - 1
src/llama_cookbook/model_checkpointing/__init__.py

@@ -1,7 +1,7 @@
 # Copyright (c) Meta Platforms, Inc. and affiliates.
 # This software may be used and distributed according to the terms of the Llama 2 Community License Agreement.
 
-from llama_recipes.model_checkpointing.checkpoint_handler import (
+from llama_cookbook.model_checkpointing.checkpoint_handler import (
     load_model_checkpoint,
     save_fsdp_model_checkpoint_full,
     save_peft_checkpoint,