diff --git a/torchtrain/checkpoint.py b/torchtrain/checkpoint.py
index 225b6712..bd141142 100644
--- a/torchtrain/checkpoint.py
+++ b/torchtrain/checkpoint.py
@@ -1,3 +1,6 @@
+# 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.
+
 import enum
 import os
 import re