@@ -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.
+
from unittest.mock import patch
import importlib
import torch
from llama_recipes.utils.train_utils import train