Update tests.py

rasbt · Oct 23, 2024 · 4f9c9fb · 4f9c9fb
1 parent ef40181
commit 4f9c9fb
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/ch05/07_gpt_to_llama/tests/tests.py b/ch05/07_gpt_to_llama/tests/tests.py
@@ -1,3 +1,10 @@
+# Copyright (c) Sebastian Raschka under Apache License 2.0 (see LICENSE.txt).
+# Source for "Build a Large Language Model From Scratch"
+#   - https://www.manning.com/books/build-a-large-language-model-from-scratch
+# Code: https://github.com/rasbt/LLMs-from-scratch
+
+# File for internal use (unit tests)
+
 import io
 import os
 import sys
@@ -8,14 +15,6 @@
 from transformers.models.llama.modeling_llama import LlamaRotaryEmbedding, apply_rotary_pos_emb
 
 
-# Copyright (c) Sebastian Raschka under Apache License 2.0 (see LICENSE.txt).
-# Source for "Build a Large Language Model From Scratch"
-#   - https://www.manning.com/books/build-a-large-language-model-from-scratch
-# Code: https://github.com/rasbt/LLMs-from-scratch
-
-# File for internal use (unit tests)
-
-
 @pytest.fixture(scope="module")
 def notebook():
     def import_definitions_from_notebook(notebooks):