diff --git a/src/open_r1/__init__.py b/src/open_r1/__init__.py
index db662173..8bb0c54c 100644
--- a/src/open_r1/__init__.py
+++ b/src/open_r1/__init__.py
@@ -1,3 +1,17 @@
+# Copyright 2025 The HuggingFace Team. 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.
+
 __version__ = "0.3.0.dev0"
 
 from .configs import DataArguments, DPOConfig, H4ArgumentParser, ModelArguments, SFTConfig
diff --git a/src/open_r1/evaluate.py b/src/open_r1/evaluate.py
index 18f5792c..ef3089ff 100644
--- a/src/open_r1/evaluate.py
+++ b/src/open_r1/evaluate.py
@@ -11,6 +11,7 @@
 # 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.
+
 """Custom evaluation tasks for LightEval."""
 
 from lighteval.metrics.dynamic_metrics import (
diff --git a/src/open_r1/generate.py b/src/open_r1/generate.py
index 7e8adc29..b2a0d7d1 100644
--- a/src/open_r1/generate.py
+++ b/src/open_r1/generate.py
@@ -1,3 +1,17 @@
+# Copyright 2025 The HuggingFace Team. 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.
+
 from typing import Optional
 
 from distilabel.llms import OpenAILLM
diff --git a/src/open_r1/sft.py b/src/open_r1/sft.py
index ac8c2de3..5df45001 100644
--- a/src/open_r1/sft.py
+++ b/src/open_r1/sft.py
@@ -1,6 +1,4 @@
-#!/usr/bin/env python
-# coding=utf-8
-# Copyright 2025 The HuggingFace Inc. team. All rights reserved.
+# Copyright 2025 The HuggingFace Team. 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.
@@ -13,6 +11,7 @@
 # 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.
+
 """
 Supervised fine-tuning script for decoder language models.