소스 검색

update policies

Sanyam Bhutani 4 달 전
부모
커밋
82340e23c8
1개의 변경된 파일4개의 추가작업 그리고 4개의 파일을 삭제
  1. 4 4
      src/llama_cookbook/policies/__init__.py

+ 4 - 4
src/llama_cookbook/policies/__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.policies.mixed_precision import *
-from llama_recipes.policies.wrapping import *
-from llama_recipes.policies.activation_checkpointing_functions import apply_fsdp_checkpointing
-from llama_recipes.policies.anyprecision_optimizer import AnyPrecisionAdamW
+from llama_cookbook.policies.mixed_precision import *
+from llama_cookbook.policies.wrapping import *
+from llama_cookbook.policies.activation_checkpointing_functions import apply_fsdp_checkpointing
+from llama_cookbook.policies.anyprecision_optimizer import AnyPrecisionAdamW