From 2ea8b7e51897be03f6657941fda0309b2d4f8220 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Sun, 29 Dec 2024 16:18:05 -0500
Subject: [PATCH] add license block

---
 src/axolotl/core/trainer_builder.py                | 14 ++++++++++++++
 src/axolotl/core/trainers/kd/__init__.py           | 14 ++++++++++++++
 .../core/trainers/kd/topk_logprob/forward_kl.py    | 14 ++++++++++++++
 3 files changed, 42 insertions(+)

diff --git a/src/axolotl/core/trainer_builder.py b/src/axolotl/core/trainer_builder.py
index 43ebf3170..def1d7a26 100755
--- a/src/axolotl/core/trainer_builder.py
+++ b/src/axolotl/core/trainer_builder.py
@@ -1,3 +1,17 @@
+# Copyright 2024 Axolotl AI. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 # pylint: disable=too-many-lines
 """
 Builder for the training args and trainer
diff --git a/src/axolotl/core/trainers/kd/__init__.py b/src/axolotl/core/trainers/kd/__init__.py
index 4add1ccc3..605a52f9b 100644
--- a/src/axolotl/core/trainers/kd/__init__.py
+++ b/src/axolotl/core/trainers/kd/__init__.py
@@ -1,3 +1,17 @@
+# Copyright 2024 Axolotl AI. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """
 KD trainer
 """
diff --git a/src/axolotl/core/trainers/kd/topk_logprob/forward_kl.py b/src/axolotl/core/trainers/kd/topk_logprob/forward_kl.py
index c9fb23c50..b050e066f 100644
--- a/src/axolotl/core/trainers/kd/topk_logprob/forward_kl.py
+++ b/src/axolotl/core/trainers/kd/topk_logprob/forward_kl.py
@@ -1,3 +1,17 @@
+# Copyright 2024 Axolotl AI. All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
 """
 loss for top_k KL divergence
 """